mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 13:50:57 +09:00
parent 5b603cecda
commit e6c8e3b107
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

4
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -1130,13 +1130,17 @@ def run(rank, world_size, args, wb=None):
if params.multi_optim: if params.multi_optim:
logging.info("Using seperate optimizers over encoder, decoder ...") logging.info("Using seperate optimizers over encoder, decoder ...")
enc_param = [] enc_param = []
enc_names = []
dec_names = [] dec_names = []
dec_param = [] dec_param = []
for n, p in model.named_parameters(): for n, p in model.named_parameters():
name = n.split('.')[1] name = n.split('.')[1]
if name == 'encoder' and 'feature_extractor' not in n: if name == 'encoder' and 'feature_extractor' not in n:
enc_names.append(n)
enc_param.append(p) enc_param.append(p)
elif 'feature_extractor' not in n: elif 'feature_extractor' not in n:
dec_names.append(n) dec_names.append(n)