diff --git a/configs/rhythm.yaml b/configs/rhythm.yaml index 830a4d8..16bd406 100644 --- 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