mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in lambda

Browse Source
This commit is contained in:
Daniel Povey 2022-04-05 13:31:52 +08:00
parent ed8eba91e1
commit d1a669162c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -784,7 +784,7 @@ def run(rank, world_size, args):
eps=1e-9, weight_decay=3e-04, target_rms=0.1) eps=1e-9, weight_decay=3e-04, target_rms=0.1)
scheduler = torch.optim.lr_scheduler.LambdaLR( scheduler = torch.optim.lr_scheduler.LambdaLR(
optimizer, optimizer,
lambda step: (params.lr_num_steps/(step + params.lr_num_steps) ** params.lr_power)) lambda step: ((params.lr_num_steps/(step + params.lr_num_steps)) ** params.lr_power))
if checkpoints and "optimizer" in checkpoints: if checkpoints and "optimizer" in checkpoints: