From c4066b7eb4f0f54b4d5e846c2712c779288dd4e1 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 21:18:15 +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 5622d3909b9c1c2fe4aee018ae79892e311468e5..93f12608000b7726f3ade325ad471a7c3783ae64 100644 GIT binary patch delta 89 zcmZozz}m2YRW!*U%+puFLeG!^2m}}y&IRVDlx`G#Z_jGV%fPU9GN(f{?+YO75)fYm niEJ)(u;HD&$=8!FGq0euB-$WWuQ;2xF delta 90 zcmZozz}m2YRW!*U%+puFLeG!^2m}}yssr;=ayE*-w`aBDWnkDenbRSf_brfh9f+@i nL^c;X*zmFzDFDIbwZ5L5IjMOXnRx}JCCw~;+gbb=|1bgoK3y44 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 3756a8d43..2b5aba9e1 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('-'*20) - print(len(input), output.size()) + print(input[0].size(), output.size()) print('-'*20) lora_out = self.lora(input)