From 4d82da2317d1e9ca4c8e1af7a6af9ad230296d16 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Fri, 3 Feb 2023 02:25:59 +0900 Subject: [PATCH] from local --- egs/aishell/ASR/transformer_ctc/.train.py.swp | Bin 53248 -> 53248 bytes egs/aishell/ASR/transformer_ctc/train.py | 4 ++-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index e6e0227cb8a8880fef11f9bba2fa2744d02e56ae..6413ab706b4ddca220f7a7aef4e8849c2cf511e2 100644 GIT binary patch delta 169 zcmZozz}&EaSuDvQ%+puFQqO<^2m}}y0$$xt{{3a6*t_+yJ9!uw6nP=S!XRlNILXhz za0H0g0&x=%Hv(}o5Gw<*5)gmoV_-N4#8ZIS7KoLB_$|Q3J`ziV_-N8#M6M-9*9+e_&v~!%|JW}h--kj7>JcOCvG^v pGx_f-uFaZzyP1S?@{5c0Qc}w@lT#IJZ50fx7$zsIu~>9K0006ZCu9Hs 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: