diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index a534691ff..cbb0b3b8a 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index aa8940f0e..a3109aa97 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1082,7 +1082,7 @@ def train_one_epoch( f"grad_scale is too small, exiting: {cur_grad_scale}" ) - if batch_idx % params.log_interval == 0: + if batch_idx % (params.log_interval*params.accum_grads) == 0: if params.multi_optim: cur_enc_lr = scheduler_enc.get_last_lr()[0] cur_dec_lr = scheduler_dec.get_last_lr()[0]