From aaf77797f3b3914ecf80690be179049fd1415a9e Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Thu, 25 May 2023 21:12:57 +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 b460695739b3a3a259d384e579ef8c33a22e8bd9..93221b3f25961e243bc26d305ece0e8beb33d38f 100644
GIT binary patch
delta 70
zcmZozz}m2YRW!*U%+puFLeG!^2m}}y+ynAcY&MF%w`VQjWnj27nbRShm4Oe)+MMc8
Y!#nx7ui)hOzJ|@pe%qD(81FIy0F>+$Qvd(}

delta 74
zcmZozz}m2YRW!*U%+puFLeG!^2m}}yn*H-rj5mtDw`a}eWnj2DnbRSh^)HCEIn|+t
bck(G;zR90_4Vj}2Vw;uywk!KF-em*;3=<Y+

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 bd2dbe4b0..f58b40ad7 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
@@ -138,7 +138,7 @@ class LoRAHook():
                            lora_alpha=1.0,
                     )
     def hook_fn(self, module, input, output):
-        print(module, input, output[0].size())
+        print(module, input, output.size())
         lora_out = self.lora(input)
         output += lora_out