From 6b4b7859db02b7d8687c5a1e725527632249fbfd Mon Sep 17 00:00:00 2001
From: xuanyoya <xuany@stanford.edu>
Date: Fri, 11 Jan 2019 15:49:04 -0800
Subject: [PATCH] [run_optimizer] update run_optimizer
---
tools/run_optimizer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tools/run_optimizer.py b/tools/run_optimizer.py
index 7fac7fe..0fa174b 100644
--- a/tools/run_optimizer.py
+++ b/tools/run_optimizer.py
@@ -29,7 +29,7 @@ def mem_explore_optimizer(arch_info, network_info, schedule_info, verbose=False)
folder = os.path.join(cwd, "dataset")
if not os.path.isdir(folder) or not os.path.exists(folder):
os.mkdir(folder)
- output_filename = os.path.join(folder, network_info['layer_name'] + '_128.csv')
+ output_filename = os.path.join(folder, network_info['layer_name'] + '.csv')
explore_points = arch_info["explore_points"]
energy_list = np.zeros(tuple(explore_points))
summary_array = np.zeros([np.product(explore_points), 12])
--
GitLab