From 66fd9d3d2825bbeae4b921e6ee0f42ddc72ba707 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sat, 30 Jan 2021 18:40:23 -0600
Subject: [PATCH] mobilenet -> mobilenet_cifar10
---
hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp | 2 +-
.../dnn_benchmarks/benchmarks/mobilenet/mobilenet_cudnn.cpp | 2 +-
hpvm/test/dnn_benchmarks/scripts/run_dnns.py | 2 +-
3 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp b/hpvm/test/dnn_benchmarks/benchmarks/mobilenet/mobilenet.cpp
index 95c10fb906..5ea5c298bf 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 6474c9f457..b7e0a71459 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 a82e580de4..0de85c7847 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":
--
GitLab