diff --git a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp index 95c10fb90618454d3da4db0cb1320331dbe5f628..5ea5c298bf7b5858af024aff7a4ee81c4b8a6ed2 100644 --- a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp +++ b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp @@ -1966,7 +1966,7 @@ typedef struct __attribute__((__packed__)) { } RootIn; int main() { - std::string dir_prefix = std::string(MODEL_PARAMS_DIR) + "/mobilenet/"; + std::string dir_prefix = std::string(MODEL_PARAMS_DIR) + "/mobilenet_cifar10/"; std::string input_path = dir_prefix + std::string("tune_input.bin"); std::string labels_path = dir_prefix + std::string("tune_labels.bin"); diff --git a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet_cudnn.cpp b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet_cudnn.cpp index 6474c9f457e1d22775d8b9ef113bbf154e451679..b7e0a714590418414a2647474526a1fb0c09e390 100644 --- a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet_cudnn.cpp +++ b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet_cudnn.cpp @@ -1967,7 +1967,7 @@ typedef struct __attribute__((__packed__)) { int main() { - std::string dir_prefix = std::string(MODEL_PARAMS_DIR) + "/mobilenet/"; + std::string dir_prefix = std::string(MODEL_PARAMS_DIR) + "/mobilenet_cifar10/"; std::string input_path = dir_prefix + std::string("tune_input.bin"); std::string labels_path = dir_prefix + std::string("tune_labels.bin"); diff --git a/hpvm/test/dnn_benchmarks/scripts/run_dnns.py b/hpvm/test/dnn_benchmarks/scripts/run_dnns.py index a82e580de4e05209fef8251d064a32a784f5f82c..0de85c7847309532db985d95aedbba02f2715059 100644 --- a/hpvm/test/dnn_benchmarks/scripts/run_dnns.py +++ b/hpvm/test/dnn_benchmarks/scripts/run_dnns.py @@ -1,7 +1,7 @@ import os import sys -dnns = ["alexnet", "alexnet2", "vgg16_cifar10", "vgg16_cifar100", "resnet18", "mobilenet", "alexnet_imagenet", "resnet50_imagenet", "vgg16_imagenet", "lenet_mnist"] +dnns = ["alexnet", "alexnet2", "vgg16_cifar10", "vgg16_cifar100", "resnet18", "mobilenet_cifar10", "alexnet_imagenet", "resnet50_imagenet", "vgg16_imagenet", "lenet_mnist"] #dnns = ["resnet50_imagenet","alexnet"] #if sys.argv[1] == "--runtime":