提交 90eea482 编写于 作者: 吴承辉

Merge branch 'skip_validate' into 'master'

support skip validation in 'all' mode

See merge request !313
...@@ -237,6 +237,7 @@ def main(unused_args): ...@@ -237,6 +237,7 @@ def main(unused_args):
# Transfer params by environment # Transfer params by environment
os.environ["MODEL_TAG"] = model_name os.environ["MODEL_TAG"] = model_name
print '=======================', model_name, '=======================' print '=======================', model_name, '======================='
skip_validation = configs["models"][model_name]["skip_validation"]
model_config = configs["models"][model_name] model_config = configs["models"][model_name]
for key in model_config: for key in model_config:
if key in ['input_nodes', 'output_nodes'] and isinstance( if key in ['input_nodes', 'output_nodes'] and isinstance(
...@@ -294,7 +295,7 @@ def main(unused_args): ...@@ -294,7 +295,7 @@ def main(unused_args):
if FLAGS.mode == "benchmark": if FLAGS.mode == "benchmark":
benchmark_model(target_soc, model_output_dir, option_args) benchmark_model(target_soc, model_output_dir, option_args)
if FLAGS.mode == "validate" or FLAGS.mode == "all": if FLAGS.mode == "validate" or (FLAGS.mode == "all" and skip_validation == 0):
validate_model(target_soc, model_output_dir) validate_model(target_soc, model_output_dir)
if FLAGS.mode == "build" or FLAGS.mode == "merge" or FLAGS.mode == "all": if FLAGS.mode == "build" or FLAGS.mode == "merge" or FLAGS.mode == "all":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册