From 5d9095354643e276703c01a7ef231835708ad10a Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 21:15:00 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 86016 -> 86016 bytes .../train_lora.py | 3 ++- 2 files changed, 2 insertions(+), 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 e85609dd784a004d50ae198bd93d7b48f28708d6..4db74ba9e3ef86479ac20701594259a451fd1b56 100644 GIT binary patch delta 105 zcmZozz}m2Yb;EmRMuyEFm>qZ+MK>D?T-N6;=4D`b48#lq3=E=^1s(i3IrtbD*n#-* z=1hkw-pQYRRg@VNprD{AGp|G=Gq0euL`NaNv;;`&6=znZYG^iV`EA$oW4y))0Nm{z AlK=n! delta 101 zcmZozz}m2Yb;EmR#?Z|lm>qZ+MK&7>T-N7p=Vf5{48#lq3=ATZ1s(i3#rYs2hc;(A zRPj!};LFNgoSKuSmy=(VsF6AOv9HSHdA=On1x1;8B^sG|1*Ij;T7KKL{1~q>0ssv6 BAK?H1 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 29cdb9b65..13acf1473 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,8 @@ class LoRAHook(): lora_alpha=1.0, ) def hook_fn(self, module, input, output): - print(input) + print(input, output.size()) + lora_out = self.lora(input) output += lora_out