repair a var

5 years ago · 94e8c3b66d
parent af8740330b
commit 94e8c3b66d
1 changed files with 2 additions and 2 deletions
--- 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()