提交 b13f3aab 编写于 作者: Y yejianwu

fix benchmark error for multi models

上级 859fbc90
...@@ -369,7 +369,8 @@ def process_models(project_name, configs, embed_model_data, vlog_level, ...@@ -369,7 +369,8 @@ def process_models(project_name, configs, embed_model_data, vlog_level,
model_config["input_shapes"], model_config["input_shapes"],
input_file_list) input_file_list)
if FLAGS.mode == "build" or FLAGS.mode == "all": if FLAGS.mode == "build" or FLAGS.mode == "benchmark" or \
FLAGS.mode == "all":
sh_commands.gen_model_code( sh_commands.gen_model_code(
"mace/codegen/models/%s" % model_name, "mace/codegen/models/%s" % model_name,
model_config["platform"], model_config["platform"],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册