feat: add precision in save_detail, manage total_batch_size param

DimensionSTP · May 11, 2024 · 5a4a1e1 · 5a4a1e1
1 parent dc49c87
commit 5a4a1e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/configs/rhythm.yaml b/configs/rhythm.yaml
@@ -57,7 +57,8 @@ 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}
-save_detail: backbone${backbone}-rnn_type${rnn_type}-rnn_num_layers${rnn_num_layers}-direction${direction}-bs${batch_size}x${devices}x${accumulate_grad_batches}
+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