未验证 提交 94e8c3b6 编写于 作者: S shaohua.zhang 提交者: GitHub

repair a var

上级 af874033
...@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册