From 11471bad386cdf3b1b1933af9db68d5f0d675432 Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Thu, 25 May 2023 21:49:52 +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 a37660fd4cea140ccb4478f7c0e309da750c4e30..5419cbebe05a56795ada6ae689096cdac0c1b4cb 100644
GIT binary patch
delta 563
zcmWN}O(??w9LDkAmj7n67s)Q(Yh#5pH{)P!yC8Wfs~zm1rI5BLLfS;Ju~IwiAm!yW
z<)Cr)QVxVD2gSilTS<`>CkNlBK2Oi-iEb&;EyXhA@O5<8v@|sfDHa*=hk_RimLrSm
zt6j0HMLa2;OJ$8znnfy5jw_SMKH?ZhC#sP*ik#vE$5_ND`p^n1z6~ND$l)4C*hUON
z1TY9M9H@fEAQ5>l5_!WbUXaE*<}r+3G@=eh{OLuqxIr4b*g*_o1ki;B)WU`$<aHvC
zI7bo*tmq;lA%+ib)Z@O8b6n#P8(2mdWhjLizgm$eTq1)hgfWJGIAKQ#avG6mJmL(S
zSjEgZHHiuM(FPk#D1;WDipU*~v4=R8Fo#*VV22hz1tJefB7rqbVgR0kgFCA_<@BnI
RNyoI3nQ>?JDMyd$_8+_YR%QSI

delta 564
zcmWm6ODIGE7{&4LUhbGX3>L<-JZ8c|i9#fg1xidP$!v&_OhyV>xr6a2WuuG*n#Cw9
zkA>8rW+Rl66bp|`p%h_d>+h@IIcN2G1DZFW*?NmQ8lBnYg~h@piX^#4`ohz;1Do<y
zquEuFYQJ8qIE>T;krX83EM8<Aa~MQD((!2(ImQt}m_k3AQHo4_#EHBif^+O)9X@o!
zg-+C<0BKO-#E_R*ktaOj0YNNb5<O@_K5}5juSMh;7YJe#8}Pvm7aEX<Y-GWNPhI2|
zVJu@2v$`QN#Arhma&cwmInJ?*Rm{K*8xml_PmIVNPO*bgxG?}HN??Z-5tGP0ZgGM&
z%wuer@Zi7OQHCt&h=PiDP2>_GY+()tCNPdl*rDP(TI2@HSi}N6Xhn7O{<TB(SJWxi
S!=^gr)w0s%%<wm~D4{>7)l-T9

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 64e534410..253daf143 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
@@ -148,7 +148,7 @@ class LoRAHook():
     def save_checkpoint(self, i, iter_, save_dir):
         if isinstance(self.lora, DDP):
             lora = self.lora.module
-        torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")
+        torch.save(lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")
 
 
 def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: