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

repair a var

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