from local

This commit is contained in:
dohe0342 2022-12-10 16:04:58 +09:00
parent 6c80a64370
commit 039cf91ca6
3 changed files with 3 additions and 0 deletions

View File

@ -1222,6 +1222,9 @@ def run(rank, world_size, args, wb=None):
if name == 'encoder' and 'feature_extractor' not in n:
enc_names.append(n)
enc_param.append(p)
elif 'ctc_output' in n:
enc_names.append(n)
enc_param.append(p)
elif 'feature_extractor' not in n:
dec_names.append(n)
dec_param.append(p)