From 4d541eb364fc63888122e5b10b27b441c7be50b3 Mon Sep 17 00:00:00 2001 From: "shaohua.zhang" 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,\