Skip to content
Snippets Groups Projects
Commit 1f4b87e9 authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Fixing VGG6_Imagenet paths

parent 1ca6f0a2
No related branches found
No related tags found
No related merge requests found
......@@ -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;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment