From da288af854169367ac70c065005471c414e32f82 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Fri, 13 Jan 2023 16:51:44 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 53248 -> 57344 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 e0e4e5e8135a4564692f5242f82f91ff920e15c4..97a98b82ada9f6d913c8f0a394f95c5b7425883b 100644
GIT binary patch
delta 982
zcmXZaUr1A77{~GFc&2V^n@-4(bd9J4=VaJLjWR4ED^MaMLW|bc8bujKQ9-bQb<s@<
zlm0Ma6Uz*oG?!l4$}58k3Zexi%@s8xqU}Pg=+F05FMQ7Xob$ft<#2vaD5i!Us+C^X
z-W{!WcY{k9<szre)4qjdqVjW@VoIvzW<@cFt=45$v$oFg|C=P1tg_Bn#_hU3bB?w3
z7M}AuoA0g*YIjRf2OH+tF$53Vuo_i(XA*ghX$&ERPISPHm6%^5lEXdp;tB$&Mg_hZ
zMHZ391TGsTC_V}=Ho}G)d@_hkp%1+{ihXEAK`)ZWE2I&}Mf6}B4AA3QsmL(;5x^!i
zp%m#7ktlAI1Vye>^x!ylq8iIFTP*Sh*Kid+G@zj41CYZD#Bl*<&<q_^WL1$d44?~6
z)Z<5y$aiFLr-n-u0UW^wd{IQ+Bc~)kCf1t5n^U&pdiz;#lU5o2R&6J{{K<o<L2K~P
zVg4zbM>tta)&D>A#WD^;7Pk<<DpX*eoM&+xQCva*Td@LKAN(Zq4~SzF=g^Mba3Dk8
zFHPikpJECToJT8au?B)H!Ovj~A?!groX{F$fxIW7HN*fS=*D)mz=)^h{sjFv2CWwy
zC^C?rLh+cuBk<t>_M-(ZSdby~w-`hWo#;R<3S>Qr35-JPhi){(4ILhl`wWJ03LZ2f
l?<DB2cz_gc;sktfpbo!@W*j4G;`uKn?AfAE<`cV2e*g(zo8$lh

delta 504
zcmXBR&ntrg7{Kx8^$z>>&UV>O)@BFgz;KvdWTZ%PP;Qz?R+hreWs}rgM4N;B$ZCg`
z5_>P^u<YU=kb{~Ug(>5p?BvH+p87mJ^*o2C=ZRmc@gvna?VlJP^#t4fLfA#tti_pp
z?#%hxq*!#-t|`h=(&2FFLymLXcG;8j7`C)}MqiZwdAg^8SjZ+4L<f9$(nPYD#W>pF
zM#U;p#uc(y#}vZoMJwJcB30y(LK0DU;lj6B<O?NSVACuyiBQZT1Rq-PZW1YC2PrII
z5&_hVBA>WJ2{~*aff3lC;iggK7`upK0R6C{)F6_<VM9#hfFglKjKK>J9yr!L_OXWu
zx=}N*0IIk}4y#zfAWSggSryTd#xlCl3BgaD$RkdgNf|^jk3M|xpuZ<o6g_<%wkCTD
KZarK`JAMKAN>z0L

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
index e17a5b85b..80b46362c 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
@@ -923,7 +923,7 @@ def train_one_epoch(
         params.best_train_loss = params.train_loss
 
 
-def run(rank, world_size, args):
+def run(rank, world_size, args, wb=None):
     """
     Args:
       rank: