From fca9a1282105fac681e3c7ba6f1ec14dec13e3a9 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 13:38:35 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 90112 -> 90112 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 ac76f89bd99e9b868dce9d11720eb8dd8c7954a0..81757c7c8210c0ec86e076f2f693881934e2f94d 100644 GIT binary patch delta 75 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}y)=$YzdC(}jRh03)JgYBI%ym1b0^@x~&J|1y b4AnsFI{l*}W6Ja}CB}Tl;_XM37_Dsq0HPEv delta 73 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}y8m8o?ENv9sD$4j?p4AU1=C++vf$=^g=Sn69 Zh8iGtoBmOeF=cv)5@W&kqe_g{wgAS`6H@>H 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 634976c3b..456ff2e4f 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 @@ -1593,7 +1593,7 @@ def run_adapter(rank, world_size, args, wb=None): lora_modules = [] for modules in model.modules(): - if isinstance(module, fairseq.modules.multihead_attention.MultiheadAttention): + if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention): for module in modules.modules(): lora_modules.append(LoRAHook(module))