From 94e8c3b66d61dd957d43edacf24377aa20acf718 Mon Sep 17 00:00:00 2001 From: "shaohua.zhang" Date: Thu, 23 Jul 2020 14:04:24 +0800 Subject: [PATCH] repair a var --- tools/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/train.py b/tools/train.py index 14949012..0f5e9039 100755 --- a/tools/train.py +++ b/tools/train.py @@ -92,7 +92,7 @@ def main(): 'fetch_name_list':eval_fetch_name_list,\ 'fetch_varname_list':eval_fetch_varname_list} - if contain_det: + if train_alg_type == 'det': program.train_eval_det_run(config, exe, train_info_dict, eval_info_dict) else: program.train_eval_rec_run(config, exe, train_info_dict, eval_info_dict) @@ -117,6 +117,6 @@ def test_reader(): if __name__ == '__main__': - startup_program, train_program, place, config, contain_det = program.preprocess() + startup_program, train_program, place, config, train_alg_type = program.preprocess() main() # test_reader() -- GitLab