From 9126cb6fb95cba31b1bd7b6a6c5e66bbc2458e5a Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 13 May 2020 16:51:36 +0800 Subject: [PATCH] modify rec saved model dir --- configs/rec/rec_mv3_none_bilstm_ctc.yml | 2 +- configs/rec/rec_mv3_none_none_ctc.yml | 2 +- configs/rec/rec_mv3_tps_bilstm_attn.yml | 2 +- configs/rec/rec_mv3_tps_bilstm_ctc.yml | 2 +- configs/rec/rec_r34_vd_none_bilstm_ctc.yml | 2 +- configs/rec/rec_r34_vd_none_none_ctc.yml | 2 +- configs/rec/rec_r34_vd_tps_bilstm_attn.yml | 2 +- configs/rec/rec_r34_vd_tps_bilstm_ctc.yml | 2 +- tools/program.py | 4 +--- 9 files changed, 9 insertions(+), 11 deletions(-) diff --git a/configs/rec/rec_mv3_none_bilstm_ctc.yml b/configs/rec/rec_mv3_none_bilstm_ctc.yml index ea104638..11a09ee9 100755 --- a/configs/rec/rec_mv3_none_bilstm_ctc.yml +++ b/configs/rec/rec_mv3_none_bilstm_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_CRNN save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_mv3_none_none_ctc.yml b/configs/rec/rec_mv3_none_none_ctc.yml index 06622783..bbbb6d1f 100755 --- a/configs/rec/rec_mv3_none_none_ctc.yml +++ b/configs/rec/rec_mv3_none_none_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_Rosetta save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_mv3_tps_bilstm_attn.yml b/configs/rec/rec_mv3_tps_bilstm_attn.yml index f5ba60d9..03a2e901 100755 --- a/configs/rec/rec_mv3_tps_bilstm_attn.yml +++ b/configs/rec/rec_mv3_tps_bilstm_attn.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_RARE save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_mv3_tps_bilstm_ctc.yml b/configs/rec/rec_mv3_tps_bilstm_ctc.yml index 132320bc..47247b72 100755 --- a/configs/rec/rec_mv3_tps_bilstm_ctc.yml +++ b/configs/rec/rec_mv3_tps_bilstm_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_STARNet save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_r34_vd_none_bilstm_ctc.yml b/configs/rec/rec_r34_vd_none_bilstm_ctc.yml index b784c7f8..10181936 100755 --- a/configs/rec/rec_r34_vd_none_bilstm_ctc.yml +++ b/configs/rec/rec_r34_vd_none_bilstm_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_CRNN save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_r34_vd_none_none_ctc.yml b/configs/rec/rec_r34_vd_none_none_ctc.yml index 9d42e6d1..ff4c5763 100755 --- a/configs/rec/rec_r34_vd_none_none_ctc.yml +++ b/configs/rec/rec_r34_vd_none_none_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_Rosetta save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_r34_vd_tps_bilstm_attn.yml b/configs/rec/rec_r34_vd_tps_bilstm_attn.yml index a77e457b..4d96e9e7 100755 --- a/configs/rec/rec_r34_vd_tps_bilstm_attn.yml +++ b/configs/rec/rec_r34_vd_tps_bilstm_attn.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_RARE save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml b/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml index 3768aa47..844721a2 100755 --- a/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml +++ b/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml @@ -4,7 +4,7 @@ Global: epoch_num: 72 log_smooth_window: 20 print_batch_step: 10 - save_model_dir: output/rec + save_model_dir: output/rec_STARNet save_epoch_step: 3 eval_batch_step: 2000 train_batch_size_per_card: 256 diff --git a/tools/program.py b/tools/program.py index a35fe6c9..6c9e9904 100755 --- a/tools/program.py +++ b/tools/program.py @@ -284,9 +284,7 @@ def train_eval_rec_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 save_model_dir[-1] == "/": - save_model_dir = save_model_dir[:-1] - if not os.path.exists(save_model_dir + config['Global']['algorithm']): + if not os.path.exists(save_model_dir): os.makedirs(save_model_dir) train_stats = TrainingStats(log_smooth_window, ['loss', 'acc']) best_eval_acc = -1 -- GitLab