diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index e6e0227cb..6413ab706 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 96c720ea1..40373b9a3 100755 --- a/egs/aishell/ASR/transformer_ctc/train.py +++ b/egs/aishell/ASR/transformer_ctc/train.py @@ -470,8 +470,8 @@ def compute_validation_loss( tot_loss.reduce(loss.device) loss_value = tot_loss["loss"] / tot_loss["frames"] - - if loss.device == 0: + + if wb is not None: wb.log({"valid/loss": loss_value}) if params.cur_epoch < 10: