Skip to content

GitLab

Unverified Commit 94e8c3b6 authored by shaohua.zhang's avatar shaohua.zhang Committed by GitHub
Browse files

repair a var

parent af874033
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/train.py
View file @ 94e8c3b6
...@@ -92,7 +92,7 @@ def main(): ...@@ -92,7 +92,7 @@ def main():
'fetch_name_list':eval_fetch_name_list,\ 'fetch_name_list':eval_fetch_name_list,\
'fetch_varname_list':eval_fetch_varname_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) program.train_eval_det_run(config, exe, train_info_dict, eval_info_dict)
else: else:
program.train_eval_rec_run(config, exe, train_info_dict, eval_info_dict) program.train_eval_rec_run(config, exe, train_info_dict, eval_info_dict)
...@@ -117,6 +117,6 @@ def test_reader(): ...@@ -117,6 +117,6 @@ def test_reader():
if __name__ == '__main__': 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() main()
# test_reader() # test_reader()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment