diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
index dd9298aa7c9c9fb5a00bd28f1a515ef32d73aef3..677411fce08641206bfdf5b608a75247c26db06c 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
@@ -27,14 +27,19 @@ void testPipeline(){
   // FIXIT: Extend this to batch of images - currently 5 images
 
   //long int test_batch_size = 9145;
-  long int test_batch_size = 2000;
+  long int test_batch_size = 4572;
   long int H = 240;
   long int W = 300;
 
   printf("Reading input\n");
-  void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
+  /*void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
                                         float_type,
-                                        test_batch_size, 1, H, W);
+                                        test_batch_size, 1, H, W); */
+  
+  void* input = readTrainedWeights("../model_params/pipeline/dataset/calibration_4572.bin",
+                                        float_type,
+                                        test_batch_size, 1, H, W); 
+  
   printf("Reading golden output\n");
   void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/GEOM-2000.bin",
   //void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
@@ -114,12 +119,9 @@ void testPipeline(){
     // NOTE-IMP: Important to include this call always before doing dumpOutput and computePSNViolation
     hpvm_request_tensor(result, 0);
     
-    dumpOutput(result);
+    dumpOutput(result, "GEOM_calib.bin");
     
-  //void* psnr_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
-                                        //float_type,
-                                        //test_batch_size, 1, H, W);
-    computePSNRViolation(result, golden_output, 30);
+    //computePSNRViolation(result, golden_output, 30);
 
     
     dumpAccuracyNorms();