modify rec saved model dir

5 years ago · 9126cb6fb9
parent 8007a95528
commit 9126cb6fb9
9 changed files with 9 additions and 11 deletions
--- 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
--- 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
--- 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
--- 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
--- 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
--- 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
--- 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
--- 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
--- 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