From fa4e732ea9dfcc01e65ee6ffd1a0a6af004eab97 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 19:26:56 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 81920 -> 81920 bytes .../train_lora.py | 4 ++-- 2 files changed, 2 insertions(+), 2 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 fbe1dd74e0c9767fd98775b6355b51d1ed7ef9a4..0dfe727ff17a2dcb54ab534fd0f85c59ef11fbca 100644 GIT binary patch delta 63 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#hw3+DML{*9trMHz(+7{#}18ZaK=oBTnKWqN}Q PqZx?7s61WAp79C*{u2=V delta 55 zcmZo@U~On%6-_b-^Ym4)&@*HJ0s#hwrRMo5ZjGW_MHz(+7$vrA8ZaK=V^p60){fC` Jx{f{L6##K`4fX&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 684122c57..ba585f293 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 @@ -1610,8 +1610,8 @@ def run_adapter(rank, world_size, args, wb=None): # if rank == 0: logging.info(module) #print(module) if isinstance(module, torch.nn.Linear): - if rank == 0: print(module) - # lora_modules.append(LoRAHook(module)) + #if rank == 0: print(module) + lora_modules.append(LoRAHook(module)) adapter_names = [] adapter_param = []