From 43954f834bb0286e2d038c13bd47f55eb66961dd Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 19:24:09 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 81920 -> 81920 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 9aa353bef94815ead9ce0175ea2b51b6943febfd..651fa1378e0f25232ef5fb724be6696dc6bbfaf0 100644 GIT binary patch delta 72 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#gF4fFgI{YKHPqKxnPS=aM0Fi2156kv>Il-`~y bz_^WP`bHf_iRnEujN**N+r4EOw<-YuVj~e| delta 72 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#gFb@TibzDCilqKxnPSvT-7Fvv{j6kv>Il-Zst bz_^WvF=zTl9Y%@iJu-|++rwlTS1ADiTR9O+ 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 5f97256f0..23a6c663c 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 @@ -1605,7 +1605,7 @@ def run_adapter(rank, world_size, args, wb=None): lora_modules = [] for modules in model.modules(): if isinstance(modules, fairseq.modules.multihead_attention.MultiheadAttention): - lora_modules.append(LoRAHook(module)) + lora_modules.append(LoRAHook(modules)) #for module in modules.modules(): # if rank == 0: logging.info(module) #print(module)