mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix adding learning rate to tensorboard

Browse Source
This commit is contained in:
Daniel Povey 2022-04-11 14:58:15 +08:00
parent d5f9d49e53
commit 5078332088
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -695,7 +695,7 @@ def train_one_epoch(
) )
if tb_writer is not None: 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( loss_info.write_summary(
tb_writer, "train/current_", params.batch_idx_train tb_writer, "train/current_", params.batch_idx_train