From 5f9f60798b07797a2648cad52f1d38306987b8db Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Fri, 13 Jan 2023 17:23:12 +0900
Subject: [PATCH] from local

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

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 05b53746c4cfe829c773fb882df8d4bf25d7faae..0b7c95b8035de8ff3532bbb5385f189c8bb719b5 100644
GIT binary patch
delta 133
zcmZozz|ydQWr7GJ-$v2*>shCAGca&Y=G-9fs?E>9@Q06qVGj_O0I@z0|Knv~*b2ls
zK&%PG8bJJ!hk@Z75YGnU3?McIViO=X24c?5xf|wiZx-KU&&;oon3tkZl$e*TU~8*j
UU^SV~N`12Y5~0l#_7?;I0H7Eiv;Y7A

delta 123
zcmZozz|ydQWr7GJ_eRn8>sc3bGcbrx=G-9fYRS*Qzy-vo_!t<PfY=s@d4TvZF9Smr
z5Ss(B84&;HVPLoe#4CZg42Ye9*a?UofmnQV?uI$slh5Ae+swJgo|(~VGM|<DWcejR
Kn<wlq2mk;(Iveo-

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
index 15d8d8b62..58230147f 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
@@ -922,7 +922,7 @@ def train_one_epoch(
                 valid_info.write_summary(
                     tb_writer, "train/valid_", params.batch_idx_train
                 )
-            if wb is not None:
+            if wb is not None and rank == 0:
                 valid_info.reduce(loss.device)
                 wb.log({"valid/loss": valid_info["simple_loss"]
                                              +valid_info["pruned_loss"]