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:46:08 +09:00
parent b3d1fdde04
commit aa20fc3d97
2 changed files with 7 additions and 1 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.

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

@ -1143,7 +1143,13 @@ def run(rank, world_size, args, wb=None):
dec_param.append(p) dec_param.append(p)
if wb is None: if wb is None:
optimizer_enc = Eve(enc_param, lr=params.peak_enc_lr) optimizer_enc = ScaledAdam(
model.parameters(),
lr=params.base_lr,
clipping_scale=2.0,
parameters_names=parameters_names,
)
optimizer_dec = Eve(dec_param, lr=params.peak_dec_lr) optimizer_dec = Eve(dec_param, lr=params.peak_dec_lr)
else: else:
logging.info('start wandb sweep optimization...') logging.info('start wandb sweep optimization...')