From f76c10ef338f07e5c1530ba6aa445a77806c26fb Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 19:49:26 +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 050baab163e23c6587c8646998a2cd34d3e5434e..ae3da84fd9d56fa83ffd518eff04a1d5fc4c5489 100644 GIT binary patch delta 92 zcmZozz}m2YRW!*U%+puFLeG!^2m}}yKHB7`T-qr5-k#Ngmw{o!WKM@@)~g`a=1hl3 t-pSp*(vv^=N{dDt#Of7iR;6la>L}!wmK2nhL>tC7^ZIS)^<(_Z2mtE<9C82v delta 68 zcmZozz}m2YRW!*U%+puFLeG!^2m}}y-rD4+oZ2Y*-kvp-mx1BvWKM@@)>k0b=1hl3 VUPgt<>wLYMdHuHY`Z0cH1OS>D6e9ot 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 264fce49b..7a9ac43ee 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 @@ -139,7 +139,7 @@ class LoRAHook(): ) def hook_fn(self, module, input, output): #print(input) - print(module, input, output.size()) + print(module, input, output[0].size(), output[1].size()) #lora_out = self.lora(input) #output += lora_out