From 85697a73704751c589eddddf510a4387c894e61b Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 11 Apr 2023 14:21:41 +0900
Subject: [PATCH] from local

---
 .../.train_adapter.py.swp                     | Bin 81920 -> 81920 bytes
 .../train_adapter.py                          |   3 ++-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp
index 695b64d9f02218d67476b7c5975b902a4f5543c1..3361d30e6707c2cd67bc18ba83b487f828270947 100644
GIT binary patch
delta 173
zcmZo@U~On%6-zP*^Ym4))H7fJ0s#hw|8GoEewA$$vl3*?+#D#l`5mu3P~tu#L^gA)
zAY(S8-2@f}hH@Y-17dj~zRS$Oa0iGF0C52j=L4|<5Z`2CU|0{t>wq{Oh!ueNKI6oT
ts+0Fk<e4n9mUp`t6Js~ibUS-Sb}@y5#NuKG1td^XRB5$+0SjY-H2_I9CaM4c

delta 137
zcmZo@U~On%6-zP*^Ym4))H7fJ0s#gF{?{fc>kBrDSqU;`Yz`FM{EpWVDDjsOBAc;Q
zkTILlZYm1{!xSL41>)b#3=F@3_#P1V0dX%7+X3-+CI*JHKzs&>Yk}Afh(X5uVVro;
SdHa1v#wASK7qT$MTLS=Ee;-!>

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
index ef48c720e..9e7accc4e 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
@@ -1055,7 +1055,8 @@ def train_one_epoch(
                     is_training=True,
                     decode = True if batch_idx % params.decode_interval == 0 else False,
                 )
-            loss_info.reduce(loss.device)
+            try: loss_info.reduce(loss.device)
+            except: pass
 
             numel = params.world_size / (params.accum_grads * loss_info["utterances"])
             loss *= numel ## normalize loss over utts(batch size)