From 507833208868b8ed07a555437891f208274bbb3d Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 11 Apr 2022 14:58:15 +0800 Subject: [PATCH] Fix adding learning rate to tensorboard --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index c24fbe9a1..b9ea0def6 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -695,7 +695,7 @@ def train_one_epoch( ) if tb_writer is not None: - tb_writer.add_scalar("train/learning_rate", cur_lr) + tb_writer.add_scalar("train/learning_rate", cur_params.batch_idx_train) loss_info.write_summary( tb_writer, "train/current_", params.batch_idx_train