diff --git a/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp b/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp index 569793db2a60af48327bf6a6328f64104b55a3e1..9c0c980d977138a628f1c0b76354d626066d77f9 100644 --- a/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp +++ b/hpvm/test/dnn_benchmarks/hpvm-c/benchmarks/alexnet2_cifar10/alexnet2_cifar10_cudnn.cpp @@ -491,7 +491,7 @@ int main() { args->dense_1_b_bytes = 0; int batch_size = 500; - int test_input_size = 10000; + int test_input_size = 5000; int batch_count = test_input_size / batch_size; std::string input_path = dir_prefix + std::string("tune_input.bin");