diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
index f44ba60e2611daf53c6fe959b400d1e1b031804c..f95a7bda4fc581e4c40d4882304156f2420f22a5 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
@@ -78,8 +78,8 @@ void testCifarNet(){
   // NOTE: CHANGED INPUT TO STANDARDIZE
   int total_runs = 50; // FOR NOW 100;
 
-  int test_input_size = 2500; //10000;
-  int batch_size = 2500;
+  int test_input_size = 5000;
+  int batch_size = 1000;
   int batch_count = test_input_size / batch_size;
   float final_accuracy = 0.0;