From 4d541eb364fc63888122e5b10b27b441c7be50b3 Mon Sep 17 00:00:00 2001
From: "shaohua.zhang" <BeyondYourself@users.noreply.github.com>
Date: Thu, 23 Jul 2020 14:07:21 +0800
Subject: [PATCH] repair a var

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

diff --git a/tools/eval.py b/tools/eval.py
index 8f7521d7..edd84a9d 100755
--- a/tools/eval.py
+++ b/tools/eval.py
@@ -48,7 +48,7 @@ from eval_utils.eval_rec_utils import eval_rec_run
 
 
 def main():
-    startup_prog, eval_program, place, config,contain_det = program.preprocess()
+    startup_prog, eval_program, place, config, train_alg_type = program.preprocess()
     eval_build_outputs = program.build(
         config, eval_program, startup_prog, mode='test')
     eval_fetch_name_list = eval_build_outputs[1]
@@ -59,7 +59,7 @@ def main():
 
     init_model(config, eval_program, exe)
 
-    if contain_det:
+    if train_alg_type == 'det':
         eval_reader = reader_main(config=config, mode="eval")
         eval_info_dict = {'program':eval_program,\
             'reader':eval_reader,\
-- 
GitLab