提交 278df5bf 编写于 作者: 刘琦

Merge branch 'fix_benchmark_error_in_multi_models' into 'master'

fix benchmark error for multi models

See merge request !411
......@@ -369,7 +369,8 @@ def process_models(project_name, configs, embed_model_data, vlog_level,
model_config["input_shapes"],
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(
"mace/codegen/models/%s" % model_name,
model_config["platform"],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册