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 2022-12-10 13:47:50 +09:00
parent 4e629e7ddf
commit 1ce3eccb0b
2 changed files with 4 additions and 5 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.

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

@ -1144,13 +1144,12 @@ def run(rank, world_size, args, wb=None):
if wb is None:
optimizer_enc = ScaledAdam(
dec_param,
lr=params.peak_enc_lr,
clipping_scale=5.0,
parameters_names=dec_names,
enc_param,
lr=param.peak_enc_lr,
clipping_scale=2.0,
parameters_names=parameters_names,
)
optimizer_dec = Eve(dec_param, lr=params.peak_dec_lr)
else:
logging.info('start wandb sweep optimization...')
logging.info(wb.config.peak_enc_lr)