diff --git a/tools/mace_tools.py b/tools/mace_tools.py index 1858b411f9f48fe0b4d38be0901406d6bde18a59..6c37fa3bc40f0ce0af1d30ce9952a0e80af6cf5e 100644 --- a/tools/mace_tools.py +++ b/tools/mace_tools.py @@ -237,6 +237,7 @@ def main(unused_args): # Transfer params by environment os.environ["MODEL_TAG"] = model_name print '=======================', model_name, '=======================' + skip_validate = configs["models"][model_name]["skip_validate"] model_config = configs["models"][model_name] for key in model_config: if key in ['input_nodes', 'output_nodes'] and isinstance( @@ -294,7 +295,7 @@ def main(unused_args): if FLAGS.mode == "benchmark": 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_validate == 0): validate_model(target_soc, model_output_dir) if FLAGS.mode == "build" or FLAGS.mode == "merge" or FLAGS.mode == "all":