From acc993cfce1982050172e6925c6d41fdbd51a815 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Fri, 3 Feb 2023 01:31:53 +0900 Subject: [PATCH] from local --- egs/aishell/ASR/transformer_ctc/.train.py.swp | Bin 49152 -> 49152 bytes egs/aishell/ASR/transformer_ctc/train.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index 6bd42da0f863a4dbbbb116fc843590ab4bb66f19..c068f352031f0ffa4f5d87a3053b93cd561b59c5 100644 GIT binary patch delta 77 zcmZo@U~Xt&7ELk;^Ym4))H7fJ0s#hwpr^N!^)`yWk7q4pWnehEnKQwek@4te-^5DR g$tTkkd2{lMi}f-~QgbylYc(eaM#^s%PJbH#0BlDXumAu6 delta 77 zcmZo@U~Xt&7ELk;^Ym4))H7fJ0s#gFm#4RrSvQKlk7q4mWneh5nKQwek@3W4-^5B* grliElN7IzJO7ctMbMlLeYc(eaM#^s%PJbH#0API>$p8QV diff --git a/egs/aishell/ASR/transformer_ctc/train.py b/egs/aishell/ASR/transformer_ctc/train.py index 7ed9e6e31..e0b612b2d 100755 --- a/egs/aishell/ASR/transformer_ctc/train.py +++ b/egs/aishell/ASR/transformer_ctc/train.py @@ -556,7 +556,7 @@ def train_one_epoch( f"batch {batch_idx}, loss[{loss_info}], " f"tot_loss[{tot_loss}], batch size: {batch_size}" ) - wb.log({"train/loss": tot_loss}) + wb.log({"train/loss": loss.item()}) if tb_writer is not None: loss_info.write_summary(