diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index 09b705645..75f9ff4c5 100644 Binary files a/egs/aishell/ASR/transformer_ctc/.train.py.swp and b/egs/aishell/ASR/transformer_ctc/.train.py.swp differ diff --git a/egs/aishell/ASR/transformer_ctc/train.py b/egs/aishell/ASR/transformer_ctc/train.py index de736b00e..41cbf7ef9 100755 --- a/egs/aishell/ASR/transformer_ctc/train.py +++ b/egs/aishell/ASR/transformer_ctc/train.py @@ -550,6 +550,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}) if batch_idx % params.log_interval == 0: