forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

repair a var

This commit is contained in:
shaohua.zhang 2020-07-23 14:04:24 +08:00 committed by GitHub
parent af8740330b
commit 94e8c3b66d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/train.py
View File

@ -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():
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()