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 663a77250..f7910a8b7 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 ff9d887c1..de80b86fc 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1038,8 +1038,8 @@ def train_one_epoch( if batch_idx % params.log_interval == 0: if params.multi_optim: - cur_enc_lr = scheduler.get_last_lr()[0] - cur_dec_lr = scheduler.get_last_lr()[0] + cur_enc_lr = scheduler_enc.get_last_lr()[0] + cur_dec_lr = scheduler_dec.get_last_lr()[0] cur_grad_scale = scaler._scale.item() if params.use_fp16 else 1.0 logging.info(