From 355e59d58b6cc2adaa143ed1f7bf09b18e3af4a3 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 20:12:12 +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 ae3da84fd9d56fa83ffd518eff04a1d5fc4c5489..1db6559fa8f286e7c2e75ddf924c6a1c4f29c762 100644 GIT binary patch delta 75 zcmZozz}m2YRW!*U%+puFLeG!^2m}}yOl|X1Zf_KQZ_gUT%fN7EGN(f{>qii4bEbno c@8s8dGSlDLF|tkG=WEq0;nV)cqMt5P*IbrkYTOA1O$q77jp%_4r=Mf@1wFaiKOXB{yB 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 7a9ac43ee..fa2cb31d0 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[0].size(), output[1].size()) + print(module, input, output) #lora_out = self.lora(input) #output += lora_out