Skip to content
Snippets Groups Projects
Commit c47c5f22 authored by Akash Kothari's avatar Akash Kothari
Browse files

Add script to run all DNNs

parent 21b5f1db
No related branches found
No related tags found
No related merge requests found
import os.path import os
from os import path import sys
#dnns = "alexnet alexnet2 vgg16_cifar10 vgg16_cifar100 resnet18 mobilenet lenet_mnist" #dnns = "alexnet alexnet2 vgg16_cifar10 vgg16_cifar100 resnet18 mobilenet lenet_mnist"
dnns = "alexnet" dnns = "alexnet"
cur_dir = "/home/akashk4/merge/profiling/hpvm/llvm/test/VISC/DNN_Benchmarks/benchmarks" cur_dir = "/home/akashk4/merge/profiling/hpvm/llvm/test/VISC/DNN_Benchmarks/benchmarks"
if __name__ == "__main__": if sys.argv[1] == "--runtime_tuning":
exec_command = "python3 run_dnn_pred.py" + " " + cur_dir + " " + dnns exec_command = "python3 run_dnn_pred.py" + " --runtime_tuning " + cur_dir + " " + dnns
print(exec_command) print(exec_command)
os.system(exec_command) os.system(exec_command)
elif sys.argv[1] == "--predictive_tuning":
exec_command = "python3 run_dnn_pred.py" + " --predictive_tuning " + cur_dir + " " + dnns
print(exec_command)
os.system(exec_command)
else:
print("Error: Use --runtime_tuning or --predictive_tuning flags!")
sys.exit()
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