From ef3051f66e9fbf95f2a4efc41870ca063946a815 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Fri, 3 Feb 2023 01:29:41 +0900 Subject: [PATCH] from local --- egs/aishell/ASR/transformer_ctc/.train.py.swp | Bin 45056 -> 49152 bytes egs/aishell/ASR/transformer_ctc/train.py | 6 ++++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index 75f9ff4c5fbc4ee8aae412ed2c9e8e91326c833e..b6b024044b293dca0576637a30906c129e4cfeeb 100644 GIT binary patch delta 590 zcmZwDPe>GD7{~Evb{C9oZD&2Kr{s0nq!6~*i`IgWAe2VnKzT4icHLQBSRK+@u?Y9z zUZkLyLv$88nBbo`D)bs;hyEc15kx_!^bmIOpl#pTwp$N;c$s-;p6B6L`5LHHg7KU+ z)PG5xO<6*jB3HkE9=-BoExsEy!d}q4XAFA}%+}aZ|C(y~d&qn5k=bn|^#?N=T~{Hi z`Aa|~7vh#4!#XzT8t1b-UO)`Lf+8C0sKQ1PZ}?5)4tk-`idRNskSP7lo(;6CMu~qG z3##N!Pp9%uxj5>mo*rdoboTP%#kp!&B<(ND8p%eBT2$l{Rxyqt{5~dfh)t|v4mP@Q z9DB_o+gQb8+(#c0IEaXBU>Q$v12)Dtn}(6ww{q zq-n~TaLdjJu?__K>T&v!&g4=hckJZtL^)e3=1=?aCNlMuM|gA7IaUAif5D5_yL{#k DXx4XY delta 356 zcmXBPu}cDR6u|NK^KK~fy0gI`2oEX?gwiEQqQR*Up+XuG9CS!*i$M^TL;VHU_vpcg_v7*2t5$nf`!_>g_8}Y6(ZYf*Ria+4aFJ`|!XiF`+Y+Q8d|TuNJzS%X zMNFeRD)Nj|G>}IeCcZ6^4z5tg9!eO)PekMiCpf@9vJpv2oX229hJpcdz 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