mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-06-09 16:50:39 +09:00
parent ca286f90cf
commit 1a88cc9dde
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp Normal file
View File

Binary file not shown.

BIN
egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/.train_tta.py.swp
View File

Binary file not shown.

10
egs/tedlium3/ASR/pruned_transducer_stateless_d2v_v2/train_tta.py
View File

@ -1515,11 +1515,11 @@ def run_pea(rank, world_size, args, wb=None):
module.lora = module.lora.to(device) module.lora = module.lora.to(device)
module.lora = DDP(module.lora, device_ids=[rank], find_unused_parameters=False) module.lora = DDP(module.lora, device_ids=[rank], find_unused_parameters=False)
for i, module in enumerate(lora_modules): for i, module in enumerate(lora_modules):
for n, p in module.lora.named_parameters(): for n, p in module.lora.named_parameters():
new_n = str(i) + n new_n = str(i) + n
pea_names.append(new_n) pea_names.append(new_n)
pea_param.append(p) pea_param.append(p)
optimizer_pea = ScaledAdam( optimizer_pea = ScaledAdam(
pea_param, pea_param,