From 067b861c702ab949cd0d9ca45c55a41c3d7e74ab Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 5 Jan 2023 14:46:15 +0800 Subject: [PATCH] Use largest LR for printing --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index dcb4a12b3..46e6b4ae8 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -963,7 +963,7 @@ def train_one_epoch( raise RuntimeError(f"grad_scale is too small, exiting: {cur_grad_scale}") if batch_idx % params.log_interval == 0: - cur_lr = scheduler.get_last_lr()[0] + cur_lr = max(scheduler.get_last_lr()) cur_grad_scale = scaler._scale.item() if params.use_fp16 else 1.0 logging.info(