From 76bb40fc3f4f585131e72b5eb598e8b84749dac8 Mon Sep 17 00:00:00 2001
From: Double_V <liuvv0203@163.com>
Date: Fri, 9 Jul 2021 14:29:39 +0800
Subject: [PATCH] fix ci

---
 tools/eval.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/eval.py b/tools/eval.py
index 8ae270b3..0120baab 100755
--- a/tools/eval.py
+++ b/tools/eval.py
@@ -55,6 +55,10 @@ def main():
 
     model = build_model(config['Architecture'])
     use_srn = config['Architecture']['algorithm'] == "SRN"
+    if "model_type" in config['Architecture'].keys():
+        model_type = config['Architecture']['model_type']
+    else:
+        model_type = None
 
     best_model_dict = init_model(config, model)
     if len(best_model_dict):
-- 
GitLab