From 15b3f9c0847f0005bed4a550c4db349b5159c9c8 Mon Sep 17 00:00:00 2001
From: DimensionSTP <ddang8jh@gmail.com>
Date: Fri, 17 May 2024 07:55:40 +0900
Subject: [PATCH] feat: a detail of ckpt name changed

---
 configs/callbacks/callbacks.yaml |  2 +-
 configs/rhythm.yaml              | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/configs/callbacks/callbacks.yaml b/configs/callbacks/callbacks.yaml
index 679fb37..2793cc1 100644
--- 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
index 16bd406..a765204 100644
--- 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}
\ No newline at end of file