diff --git a/tools/train.py b/tools/train.py index f0791eb174f385915e823feeb9a88a89d6300b49..7e13ccc4a0d5216c8b68bc950e1e45394ce48a4f 100755 --- a/tools/train.py +++ b/tools/train.py @@ -92,7 +92,7 @@ def main(): 'fetch_name_list':eval_fetch_name_list,\ 'fetch_varname_list':eval_fetch_varname_list} - if isContain_det: + if contain_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)