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

repair a var

parent 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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment