From c983ee5debfb878f1fdd3138576efc61b52d0c1a Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Thu, 25 May 2023 21:49:46 +0900
Subject: [PATCH] from local

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

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp
index 3bd2137a0047d86956229bea927cf69998e1f245..a37660fd4cea140ccb4478f7c0e309da750c4e30 100644
GIT binary patch
delta 581
zcmYk!ODIGE7{&4LUM_Rzfraa}Fic@VqL7kjQ9>xmtVB#Q84;S5>rJWIC}V+UG0Ms#
z*?1HS8Ko=~3-6Lrgq5xT(y8C+tj<J8n+Rz}cR_o7Q$}fip^yZTME79N$)vGwC|{MD
zOBJaI>Q#!<^u&v#AQ?$Gj}zI%Ec#H3G<-Tlj&O(rjH4HgC`LLy>>_V?K?FNk#SA*(
zMh7a9i&VRqqA*@<B9C~$J;GRoA6;lb4l-fEuT|t37YJh=YnXu-Zqy+g8E_&7pSs8m
zP7uI?Zi-AZ`q7GVWZ}xfIU?A`GN#~#fp}Q)6Dx9yGi+f9UU<-iBDmndON_`JZg7kh
z%$d{(d>B9*O5lW!D5!YXL@sfF4a~yC7)DVB7gT&li(Dgs1<b>T7F6JWcApG&DROGx
R^VcZ0V0E*S7aVR@0)MPNR8ar`

delta 587
zcmYk!O(??w9LDkAfBdm|o0tE=%Q6m5V!LQ_@pAAIT6sIj!LVhunwJyBf3s39N)F>>
z<%El+p<HB2IjA{!$!d|Jrj#@CJ$mZ%^qihxR1HQ|ZNfIv({FV&+J%@!V#9%{!;lu&
zl+RYxVi0jgjBQGd?k*N7Ln%sdQ6#d9ASTfU4ex~_CpbnN3m8WiY^cCnfygsbNZ<ew
z=orNi2GN3gloyCD-}xeGJmLZO*ug61F@{dmp&D8EGKoCl5<A$&7Ib(qgm%=T3YAds
zZWKx43~LA*b&+L;2YqnBit9Yik-$DSumms6FhRvfuE;IUu?IiAaHAVdupkeq9FaRD
zaf(fZbSeNJrs0AemB_}gD)NI@Tp^Aqf>^{XX3z`^ezHX}xWO92Sb-0{aN>VrY0a?t
aGGIAc@5q=Uoi3$8o%i^?jgdK*vi=9=0$BY3

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
index 692cbf639..64e534410 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
@@ -147,7 +147,7 @@ class LoRAHook():
 
     def save_checkpoint(self, i, iter_, save_dir):
         if isinstance(self.lora, DDP):
-            self.lora = self.lora.module
+            lora = self.lora.module
         torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")