From 0b286e5f160db30b26417d42bb367ed768c16f13 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Fri, 13 Jan 2023 17:18:25 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 69632 -> 69632 bytes
 .../train.py                                  |   1 +
 2 files changed, 1 insertion(+)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp
index af12aed1e4de65f609644b949ef9a004abbeb6dc..8d2ad244de6cb2dec1946701a178961363aeb3db 100644
GIT binary patch
delta 167
zcmZozz|ydQMJ&l6%+puFQqO<^2m}}yJcJJ>r|#G&W+lidwmDF6^E}?g+zbrjJP=v2
z$$}fCow)cJ7*6pqFf;+NEfDhn@nK#DhAJR72VyfI{twh~2Z&b!aTyRh1F;hjI|8vd
x&*r=hE4eqj?fK6%dCz`+p`z53(&SW)oc!Wqy_D3l%;eO`^@|lZN9<n_0043ID1ZO}

delta 175
zcmZozz|ydQMJ&l6%+puFQqO<^2m}}y7786qzP^2<n3W)-=;lDd&GUG#aWgQu@jzrn
zCkt+ncKXT3z_1aBbAZ?yh(GZ%FsuaPG$2j|Vm%<{2I3n)T}y$u2#5=TxB!SjCQ0&c
p&fBn(d-CnOqEacSAT>>)s5CDHL~Rz|^Pg#Q?_%N25&KsJ0073~EusJb

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
index 1f4f8d766..15d8d8b62 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
@@ -923,6 +923,7 @@ def train_one_epoch(
                     tb_writer, "train/valid_", params.batch_idx_train
                 )
             if wb is not None:
+                valid_info.reduce(loss.device)
                 wb.log({"valid/loss": valid_info["simple_loss"]
                                              +valid_info["pruned_loss"]
                                              +valid_info["ctc_loss"]