From eee751a5dd87e825b729a87075894eef0b2a1b01 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 19:39:18 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 86016 -> 86016 bytes .../train_lora.py | 4 ++-- 2 files changed, 2 insertions(+), 2 deletions(-) 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 bc626fd43a4fb12e3ffebcccff2e301462924860..d6bd5942762f94db10ffc32c1e833a9cae76bda9 100644 GIT binary patch delta 627 zcmXZZNhrht6vy#5nm=ZVMKh%g&w;%@-juq5s<&{oN){u_BKMq7yE3ARZqvBA3|46uME2 zNGQE?8tzMC_F@pT;TwF*u@e8=tC2ns6!!=pvSwydvJ;) z9AXS^)WLxi-0|`oSgt91Sir=trCK)<5%N}Nn&vm_b>)fa9=~tM?^Eq%s@pT{YG;%m Dd0b<} delta 628 zcmXZZO(??w7{~Ev%O5ji*_Lw9khjQ7DJ!(Kn=+D@7t^YlE2r|0klR+Yf2V(Knyt#z0!CFMf&BBK^}uQOzdT9q7~ zNN`?b>rn;6E?c2alWqP_c*W88;rFRUrsMv$#$(&$IaTBdA@radT}Z=cipUkhm_;8f zNI*OkoN{wJb{7HkqY~vPh5MnnARke!$Q@3wi3NBu zh$h%k1S6tJB5!!bB_deI9ELEEft~I(xKp8;XOFR zF^(_^7aXWUChmFpEp#`OeJo*mS7%p^sa4+akUdu$+-gu+4Q}7C$2V*)sxc3`yWIX@? 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 aa9197980..1275041fd 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 @@ -141,8 +141,8 @@ class LoRAHook(): #print(input) print('fuck!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!') print(module, input, output) - lora_out = self.lora(input) - output += lora_out + #lora_out = self.lora(input) + #output += lora_out def save_checkpoint(self, i, iter_, save_dir): torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")