From a70eb4e65ca5c2164ef11b6a344d879945b3f0ad Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 27 Dec 2022 10:32:53 +0900
Subject: [PATCH] from local

---
 egs/librispeech/ASR/.run_adapter.sh.swp       | Bin 12288 -> 12288 bytes
 .../.train.py.swp                             | Bin 106496 -> 106496 bytes
 .../train.py                                  |   9 +++++++--
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/egs/librispeech/ASR/.run_adapter.sh.swp b/egs/librispeech/ASR/.run_adapter.sh.swp
index 81d7ff2a7de24c1f3a9973cadd6688a25114d6c8..e90884387f1618f3fe008fdd6d3198a4b2b23494 100644
GIT binary patch
delta 31
lcmZojXh;xCG6?hZRj|}EU;qLE28LT+tCIDLH;TQ|2LOac2+sfj

delta 31
lcmZojXh;xCG6?hZRj|}EU;qLE1_nv5Rmq17H;TQ|2LOO62%rD}

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index a88512bad27fa5da5f1f347007669f880636b948..54667e9778a8a7a74d2302a9f61d6bba5add27a3 100644
GIT binary patch
delta 321
zcmZoTz}9epO)SYE%+puFQqO<^2m}}yT)kH%w-;{|yCuXJy7{A!uq^LkW(EdhR)}oq
zW<djaMyn-k3=E|}900_oK&%eLvOs(RsB9V#PX*#BKwJ&PQ9v9C#CAYz%(|J8<s|du
zSEkvlsX4`|R+B@_3}qC6ATv!NFTX?~u_(P*FEJ%0J~1V+pd_^jq`-c%<z;dEto+P8
z+mNEtR0fc86p&X?l%HH|Tb^H(lM-K?S(U1zz<{Cv$eFytOwJK5np>b(T##6vr%_az
ghs$aOkOLKf3Se#lnmYO8L+R%KX50UpF*0TX0P87K+5i9m

delta 196
zcmZoTz}9epO)SYE%+puFQqO<^2m}}y)Vx+DzbxD+c1wscVDm>IVOicj76yh_tPt6N
z&4LE<j9T4n3=EAx><h#`K<o{~MnEhG!~#J46{sAf<t6K8MwSki$@f2UOg1*np8V0&
z&M3D)ueczwJWr#jG%r3eC9$9+wMa)Hu_(RRR--(>C?_SpII}7hBBG;Eo}{TWIl)YX
bBP%~M&o-o}bn?QjQq8~3w*NL`WX%Ks&eb>E

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index 66f785045..eecd54f6a 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1707,9 +1707,14 @@ def main():
     world_size = args.world_size
     assert world_size >= 1
     if world_size > 1:
-        mp.spawn(run_adapter, args=(world_size, args, wb), nprocs=world_size, join=True)
+        mp.spawn(run if not args.add_adapter else run_adapter, 
+                 args=(world_size, args, wb), 
+                 nprocs=world_size, 
+                 join=True
+            )
     else:
-        run(rank=0, world_size=1, args=args, wb=wb)
+        if not args.add_adapter: run(rank=0, world_size=1, args=args, wb=wb)
+        else: run(rank=0, world_size=1, args=args, wb=wb)
 
 
 torch.set_num_threads(1)