From ed84290b9661e5d393657e3be5c7c139fb610243 Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Tue, 20 Mar 2018 20:10:55 +0800
Subject: [PATCH] compatible with skip_validation parameter

---
 tools/mace_tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/mace_tools.py b/tools/mace_tools.py
index 2f4abd28..a9aebdc9 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_validation = configs["models"][model_name]["skip_validation"]
+        skip_validation = configs["models"][model_name].get("skip_validation", 0)
         model_config = configs["models"][model_name]
         for key in model_config:
           if key in ['input_nodes', 'output_nodes'] and isinstance(
-- 
GitLab