diff --git a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_imagenet/src/vgg16_imagenet_loop.cpp b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_imagenet/src/vgg16_imagenet_loop.cpp index 44f46b96afc828db2b163dc09d0192ad57a85f1a..6020d49cee402cc6e7296d1ff05d16ad37382886 100644 --- a/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_imagenet/src/vgg16_imagenet_loop.cpp +++ b/llvm/test/VISC/DNN_Benchmarks/benchmarks/vgg16_imagenet/src/vgg16_imagenet_loop.cpp @@ -886,7 +886,8 @@ RootIn; int main(){ - std::string dir_prefix = std::string("/home/hsharif3/vgg16_imagenet/"); + + std::string dir_prefix = std::string("/home/hsharif3/vgg16_imagenet_tune/"); std::string input_path = dir_prefix + std::string("input.bin"); std::string labels_path = dir_prefix + std::string("labels.bin"); std::string conv2d_1_w_path = dir_prefix + std::string("conv2d_1_w.bin"); @@ -1027,8 +1028,8 @@ int main(){ startMemTracking(); startProfiling(); - unsigned int batch_size = 20; - unsigned int test_input_size = 500; + unsigned int batch_size = 50; + unsigned int test_input_size = 1000; unsigned int batch_count = test_input_size / batch_size;