diff --git a/tools/mace_tools.py b/tools/mace_tools.py index 6c37fa3bc40f0ce0af1d30ce9952a0e80af6cf5e..2f4abd28b82263cd6ad144a87868aab70dd0bb18 100644 --- a/tools/mace_tools.py +++ b/tools/mace_tools.py @@ -237,7 +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"] + skip_validation = configs["models"][model_name]["skip_validation"] model_config = configs["models"][model_name] for key in model_config: if key in ['input_nodes', 'output_nodes'] and isinstance( @@ -295,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" and skip_validate == 0): + if FLAGS.mode == "validate" or (FLAGS.mode == "all" and skip_validation == 0): validate_model(target_soc, model_output_dir) if FLAGS.mode == "build" or FLAGS.mode == "merge" or FLAGS.mode == "all":