Skip to content
Snippets Groups Projects
Commit b5ee183a authored by Nathan Zhao's avatar Nathan Zhao
Browse files

fix file paths

parent b4fb8153
No related branches found
No related tags found
No related merge requests found
...@@ -143,10 +143,10 @@ if __name__ == '__main__': ...@@ -143,10 +143,10 @@ if __name__ == '__main__':
# *** Below are Parameters specific to each benchmark ***** # *** Below are Parameters specific to each benchmark *****
reload_dir = MODEL_PARAMS_DIR + '/alexnet_cifar10/' reload_dir = MODEL_PARAMS_DIR + '/alexnet_cifar10/'
## Either the HPVM weights are loaded (above) or the Keras Model from the path below ## Either the HPVM weights are loaded (above) or the Keras Model from the path below
keras_model_file = MODEL_PARAMS_DIR + '/alexnet_cifar10/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/alexnet_cifar10/model.h5'
data_dir = '' # if reloading weights, data_dir can be set to empty string (value is ignored) data_dir = 'data/alexnet_cifar10_hpvm/' # if reloading weights, data_dir can be set to empty string (value is ignored)
src_dir = 'data/alexnet_cifar10_src/' # Directory where HPVM sources are downloaded src_dir = 'src/alexnet_cifar10_src_hpvm/' # Directory where HPVM sources are downloaded
num_classes = 10 # Specify num out output classes - CIFAR10 has `10` classes num_classes = 10 # Specify num out output classes - CIFAR10 has `10` classes
batch_size = 500 # Batch Size set to 500 - Adjust this value based on your GPU memory batch_size = 500 # Batch Size set to 500 - Adjust this value based on your GPU memory
......
...@@ -136,9 +136,9 @@ if __name__ == '__main__': ...@@ -136,9 +136,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/alexnet2_cifar10/' reload_dir = MODEL_PARAMS_DIR + '/alexnet2_cifar10/'
keras_model_file = MODEL_PARAMS_DIR + '/alexnet2_cifar10/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/alexnet2_cifar10/model.h5'
data_dir = '' data_dir = 'data/alexnet2_cifar10/'
src_dir = 'data/alexnet2_cifar10_src/' src_dir = 'src/alexnet2_cifar10_src/'
num_classes = 10 num_classes = 10
batch_size = 500 batch_size = 500
......
...@@ -93,9 +93,9 @@ if __name__ == '__main__': ...@@ -93,9 +93,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/alexnet_imagenet/' reload_dir = MODEL_PARAMS_DIR + '/alexnet_imagenet/'
keras_model_file = MODEL_PARAMS_DIR + '/alexnet_imagenet/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/alexnet_imagenet/model.h5'
data_dir = '' data_dir = 'data/alexnet_imagenet/'
src_dir = 'data/alexnet_imagenet_src/' src_dir = 'src/alexnet_imagenet_src/'
num_classes = 1000 num_classes = 1000
batch_size = 50 batch_size = 50
......
...@@ -102,9 +102,9 @@ if __name__ == '__main__': ...@@ -102,9 +102,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/lenet_mnist/' reload_dir = MODEL_PARAMS_DIR + '/lenet_mnist/'
keras_model_file = MODEL_PARAMS_DIR + '/lenet_mnist/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/lenet_mnist/model.h5'
data_dir = '' data_dir = 'data/lenet_mnist/'
src_dir = 'data/lenet_mnist_src/' src_dir = 'src/lenet_mnist_src/'
num_classes = 10 num_classes = 10
batch_size = 500 batch_size = 500
......
...@@ -182,9 +182,9 @@ if __name__ == '__main__': ...@@ -182,9 +182,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/mobilenet_cifar10/' reload_dir = MODEL_PARAMS_DIR + '/mobilenet_cifar10/'
keras_model_file = MODEL_PARAMS_DIR + '/mobilenet_cifar10/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/mobilenet_cifar10/model.h5'
data_dir = '' data_dir = 'data/mobilenet_cifar10/'
src_dir = 'data/mobilenet_cifar10_src/' src_dir = 'src/mobilenet_cifar10_src/'
num_classes = 10 num_classes = 10
batch_size = 500 batch_size = 500
......
...@@ -554,9 +554,9 @@ if __name__ == '__main__': ...@@ -554,9 +554,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/resnet18_cifar10/' reload_dir = MODEL_PARAMS_DIR + '/resnet18_cifar10/'
keras_model_file = MODEL_PARAMS_DIR + '/resnet18_cifar10/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/resnet18_cifar10/model.h5'
data_dir = '' data_dir = 'data/resnet18_cifar10/'
src_dir = 'data/resnet18_cifar10_src/' src_dir = 'src/resnet18_cifar10_src/'
num_classes = 10 num_classes = 10
batch_size = 500 batch_size = 500
......
...@@ -141,9 +141,9 @@ if __name__ == '__main__': ...@@ -141,9 +141,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/resnet50_imagenet/' reload_dir = MODEL_PARAMS_DIR + '/resnet50_imagenet/'
keras_model_file = MODEL_PARAMS_DIR + '/resnet50_imagenet/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/resnet50_imagenet/model.h5'
data_dir = '' data_dir = 'data/resnet50_imagenet/'
src_dir = 'data/resnet50_imagenet_src/' src_dir = 'src/resnet50_imagenet_src/'
num_classes = 1000 num_classes = 1000
batch_size = 50 batch_size = 50
......
...@@ -184,9 +184,9 @@ if __name__ == '__main__': ...@@ -184,9 +184,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/vgg16_cifar10/' reload_dir = MODEL_PARAMS_DIR + '/vgg16_cifar10/'
keras_model_file = MODEL_PARAMS_DIR + '/vgg16_cifar10/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/vgg16_cifar10/model.h5'
data_dir = '' data_dir = 'data/vgg16_cifar10/'
src_dir = 'data/vgg16_cifar10_src/' src_dir = 'src/vgg16_cifar10_src/'
num_classes = 10 num_classes = 10
batch_size = 500 batch_size = 500
......
...@@ -199,9 +199,9 @@ if __name__ == '__main__': ...@@ -199,9 +199,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/vgg16_cifar100/' reload_dir = MODEL_PARAMS_DIR + '/vgg16_cifar100/'
keras_model_file = MODEL_PARAMS_DIR + '/vgg16_cifar100/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/vgg16_cifar100/model.h5'
data_dir = '' data_dir = 'data/vgg16_cifar100/'
src_dir = 'data/vgg16_cifar100_src/' src_dir = 'src/vgg16_cifar100_src/'
num_classes = 100 num_classes = 100
batch_size = 100 batch_size = 100
......
...@@ -126,9 +126,9 @@ if __name__ == '__main__': ...@@ -126,9 +126,9 @@ if __name__ == '__main__':
### Parameters specific to each benchmark ### Parameters specific to each benchmark
reload_dir = MODEL_PARAMS_DIR + '/vgg16_imagenet/' reload_dir = MODEL_PARAMS_DIR + '/vgg16_imagenet/'
keras_model_file = MODEL_PARAMS_DIR + '/vgg16_imagenet/weights.h5' keras_model_file = MODEL_PARAMS_DIR + '/vgg16_imagenet/model.h5'
data_dir = '' data_dir = 'data/vgg16_imagenet/'
src_dir = 'data/vgg16_imagenet_src/' src_dir = 'src/vgg16_imagenet_src/'
num_classes = 1000 num_classes = 1000
batch_size = 25 batch_size = 25
......
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