From 501d129957b619dc1468ba6ee11d063ba87c5153 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 25 May 2023 11:50:11 +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 90140020255f3484a49fced888e0fbe5f9cc697b..e4c5b3fdcddebd5c894db3052bae5b69e4cd417d 100644
GIT binary patch
delta 585
zcmXxcO(??w9LDjV+26eEFY=O<@_JyVaZnDzfmkUn<DlkcCn2*eEA^k8C>vJfHX)og
ztrSiUL%9g0C}jsPS)>$2@^$dk=hxG7_<4hhH>l+HTWoFZg>_XHAsHgN+7ZW(wv+o}
zR|;4>fs|gmwxNow!-*0UqX_3lkqGADLI=w5nI>|KBSbNSQFNgZ<@iVyc|#l*IKU?S
z7=!}@u%?P8wUkmA@tPv?ghxCege6R42sYH95?RpUIa%ZqA#7s{ez@U4C#qpW1v2qz
z5V^ytK@(Y_1mMFsde90Bu9J9<3+!VRb8w>&`N+YyA`-(H_Am)Ioajaa%*aDrFLIAN
zoM@aiEa0zuFabN7P=RcuL&bZd$Q7apV;-}Zf)_0?L&cX)<Q6LkU=bcT@t=hobvfLe
apx2M~jkyAczN$#DUrp1Q>La_u*4Qtd4qXBO

delta 591
zcmXxcPbk9y7{>9J?YG&mNUbnh{uK%rWlEA9tb_wO5c!jc%@)PdwkS4MVnqp!6GdTi
z9Lhx~<)F5MKT%RdadPp!c<S@^^c-GqNb!agV|Q6=eS_IjQYIut#8htU?ROs-KROj7
zi!-R}>hx?Ri>zZ1CS)Q5mq{YK@M8$|$i`Qq$SF<`!4wA21PgNTsTX;}3$Ad45PaxC
z2ij4q7mt)uvY_BqC-Q_xJYX9Om_#2MQH*>f;Wt6#8Q0jx7B=C79UZWu2zkhb9$#9K
zD9*JWktIq1v#_BBH7LZbhUd6K7^|3p9qCAe0Y8dJ3>P@W80;8CGb&JkRJ^D~?h(bA
zhqHz`jKc{B+E9sHBtr`g-c=$uh+qePOv8l<RHFbIe8-F2VF>}u!wCoevluh1?AOJq
fgUha>;GSC_KIm2@>Z;5Gwvn+>b8+~1*!uJb3k6(S

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 cc336a812..b7abd5db6 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
@@ -140,7 +140,7 @@ class LoRAHook():
                     )
     def hook_fn(self, module, input, output):
         lora_out = self.lora(input)
-        output = input + lora_out
+        output += lora_out
 
     def save_checkpoint(self, i, iter_, save_dir):
         torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")