fix det inference bug and optimize save path

2026-09-24 23:33:08 +08:00 · 2020-05-13 16:05:00 +08:00
parent ab8fd7d947
commit e3f171a3bb
16 changed files with 57 additions and 30 deletions
@@ -219,6 +219,8 @@ def train_eval_det_run(config, exe, train_info_dict, eval_info_dict):
    eval_batch_step = config['Global']['eval_batch_step']
    save_epoch_step = config['Global']['save_epoch_step']
    save_model_dir = config['Global']['save_model_dir']
+    if not os.path.exists(save_model_dir):
+        os.makedirs(save_model_dir)
    train_stats = TrainingStats(log_smooth_window,
                                train_info_dict['fetch_name_list'])
    best_eval_hmean = -1