From d0aa7e734756b2baf39955298c503389a782797e Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 12:54:23 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 81920 -> 86016 bytes .../train_lora.py | 6 +++++- 2 files changed, 5 insertions(+), 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 ce8fed5e729bbeb22b3711c2b54d2f6212ff3e03..f6214a200be14552e8dc144f91e24e0a24eae357 100644 GIT binary patch delta 239 zcmZo@U~O2yx`9c6QFt@6z*|PfvdJF>#f3x|Ab=f8TeEHcCHRcDNnx4-BMUDpBLl-v zCZLo614G$%K?TOsj7oOQ3=Gmh%nHOTKn#-a1L7{-iZ&MZWmBu)M5kxcTX{+ delta 117 zcmZozz}nEjx`9c6QD`%>z*|Pf%E=!E#W!mR-Qb=0K%q%tngSyWuK)uBg8&nVU|^`+ zE~vnGno-e;nSntWh_xu=@7`N|XXRKw~ KzFn2kQwjh+;~HQ9 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 824557ece..316690251 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 @@ -1589,7 +1589,11 @@ def run_adapter(rank, world_size, args, wb=None): if world_size > 1: logging.info("Using DDP") model = DDP(model, device_ids=[rank], find_unused_parameters=True) - + + for module in model.modules(): + print(module) + + exit() adapter_names = [] adapter_param = [] for n, p in model.named_parameters():