feat: a detail of ckpt name changed

DimensionSTP · May 16, 2024 · 15b3f9c · 15b3f9c
1 parent 1eef4ff
commit 15b3f9c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/configs/callbacks/callbacks.yaml b/configs/callbacks/callbacks.yaml
@@ -6,7 +6,7 @@ model_checkpoint:
   save_last: True
   verbose: True
   dirpath: ${connected_dir}/checkpoints/${model_name}/${dataset_name}/${is_tuned}/${strategy}/${save_detail}
-  filename: epoch{epoch}
+  filename: epoch={epoch}
   auto_insert_metric_name: False
 
 early_stopping:

diff --git a/configs/rhythm.yaml b/configs/rhythm.yaml
@@ -57,14 +57,14 @@ hparams_save_path: ${connected_dir}/hparams/${model_name}/${dataset_name}/${num_
 tuned_hparams_path: ${hparams_save_path}/best_params.json
 
 project_name: ${model_name}-${dataset_name}-${mode}
-total_batch_size: bs${batch_size}x${devices}x${accumulate_grad_batches}
-save_detail: backbone${backbone}-rnn_type${rnn_type}-rnn_num_layers${rnn_num_layers}-direction${direction}-precision${precision}-${total_batch_size}
+total_batch_size: bs=${batch_size}x${devices}x${accumulate_grad_batches}
+save_detail: backbone=${backbone}-rnn_type=${rnn_type}-rnn_num_layers=${rnn_num_layers}-direction=${direction}-precision=${precision}-${total_batch_size}
 resumed_step: 0
-ckpt_path: ${callbacks.model_checkpoint.dirpath}/epoch${epoch}.ckpt
+ckpt_path: ${callbacks.model_checkpoint.dirpath}/epoch=${epoch}.ckpt
 
 submission_file_name: test
-logit_name: ${save_detail}-epoch${epoch}
-submission_name: ${save_detail}-epoch${epoch}
+logit_name: ${save_detail}-epoch=${epoch}
+submission_name: ${save_detail}-epoch=${epoch}
 
 run_name: ${project_name}-${mode}
 work_dir: ${hydra:runtime.cwd}