From 1c032d6d410a755f5999e6c59bbe256e6ff61058 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 12:47:56 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 45056 -> 45056 bytes .../.train_lora.py.swp | Bin 81920 -> 81920 bytes .../train_lora.py | 3 +++ 3 files changed, 3 insertions(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp index 855052f585ba4eebe35ce474dd222c152206bca1..e191309b5674aa0af0f7168af0da6471d0efb796 100644 GIT binary patch delta 33 ncmZp8z|`=7Ni4}A%+puFQqO<^2m}}y%sX;Z_6ctkd$k?_p~eZ} delta 33 ncmZp8z|`=7Ni4}A%+puFQqO<^2m}}yKGx-?baQPKd$k?_r%?)3 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 0fbbb1aeb55a3f0d9c6bf3c5c6e1143fd91eec92..64d2131a0cf900d6811188c953e7ec625ca91d7b 100644 GIT binary patch delta 144 zcmZo@U~On%-SD27kzw-(<^&!_q0NB;oAr7B@-Q$Y@-i?m2rw`RO%`;p7Q6!FZ35zr zK)eBn6F0{?u<%Zv?;GI8pa2E=r6mQWB?`6*nRy^uTOlXEC@~%=3ReLW0IDfY%}LV( P@ioAzH8+3v<@^Ky*2XEC delta 86 zcmZo@U~On%-SD27F?90><^&!_fz5#eoAr5}c^Mey12KaD5KR_zuoip| a&9M$FypyBj>?SAr25=}q!RGJ2oSy*R{T7-4 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 6ba9d5a91..1265dbd1d 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 @@ -137,6 +137,9 @@ class LoRAHook(): lora_alpha=1.0, ) def hook_fn(self, module, input, output): + print(input.size()) + lora_out = self.lora(input) + output = input + lora_out def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: