提交 ece5761c 编写于 作者: 卢旭辉

Merge branch 'fix-validate-layers' into 'master'

fix the bug of validate layers's log

See merge request !1149
...@@ -713,16 +713,16 @@ class DeviceWrapper: ...@@ -713,16 +713,16 @@ class DeviceWrapper:
) )
for runtime in runtime_list: for runtime in runtime_list:
device_type = parse_device_type(runtime) device_type = parse_device_type(runtime)
log_dir = mace_model_dir + "/" + runtime
if os.path.exists(log_dir):
sh.rm('-rf', log_dir)
os.makedirs(log_dir)
for output_config in output_configs: for output_config in output_configs:
self.run_model(flags, configs, target_abi, model_name, self.run_model(flags, configs, target_abi, model_name,
output_config, runtime, tuning) output_config, runtime, tuning)
if flags.validate: if flags.validate:
log_file = "" log_file = ""
if flags.layers != "-1": if flags.layers != "-1":
log_dir = mace_model_dir + "/" + runtime
if os.path.exists(log_dir):
sh.rm('-rf', log_dir)
os.makedirs(log_dir)
log_file = log_dir + "/log.csv" log_file = log_dir + "/log.csv"
model_file_path, weight_file_path = \ model_file_path, weight_file_path = \
get_model_files( get_model_files(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册