diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index 75f9ff4c5..b6b024044 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 41cbf7ef9..7cd7b4f30 100755 --- a/egs/aishell/ASR/transformer_ctc/train.py +++ b/egs/aishell/ASR/transformer_ctc/train.py @@ -460,6 +460,10 @@ def compute_validation_loss( tot_loss.reduce(loss.device) loss_value = tot_loss["loss"] / tot_loss["frames"] + + if loss.device == 0: + wb.log({"valid/loss": loss_value) + if params.cur_epoch < 10: params.best_valid_losses[params.cur_epoch] = loss_value @@ -552,8 +556,6 @@ def train_one_epoch( ) wb.log({"train/loss": tot_loss}) - if batch_idx % params.log_interval == 0: - if tb_writer is not None: loss_info.write_summary( tb_writer, "train/current_", params.batch_idx_train