From 447f98e2a781e584727d30a716e41e5507d4e08c Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 21:19:02 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 86016 -> 86016 bytes .../train_lora.py | 8 ++++---- 2 files changed, 4 insertions(+), 4 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 93f12608000b7726f3ade325ad471a7c3783ae64..7e7b2e29c44b1ffebbb9d457a3cf6509fc8ab729 100644 GIT binary patch delta 128 zcmZozz}m2YRW!*U%+puFLeG!^2m}}y*n;v?YB!3$w`bMoWnfr7nbRR$P?V2>;Te#3 z9*EC@L^l^YSn^JPW5>ufd6%!KTC_o|CIf>46et%IW#*M=sOze085w9IgeQOS)nim{ Nmh;;#=g0Vn5dhIkA0Pk# delta 120 zcmZozz}m2YRW!*U%+puFLeG!^2m}}y&IRVDlx`G#Z_jGV%fPU9GN(hhpco$m!wVqq z5)fYmiEb`*u;gV{&dHy=+*e6alYv123JQuc^GY<-b=9?u3^d`Qli&O5HOu*Jm-AzM G#0UTlQ613$ 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 2b5aba9e1..75b12e33a 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,11 +138,11 @@ class LoRAHook(): lora_alpha=1.0, ) def hook_fn(self, module, input, output): - print('-'*20) - print(input[0].size(), output.size()) - print('-'*20) + #print('-'*20) + #print(input[0].size(), output.size()) + #print('-'*20) - lora_out = self.lora(input) + lora_out = self.lora(input[0]) output += lora_out def save_checkpoint(self, i, iter_, save_dir):