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

Increase lr_update_period from 200,4000 to 400, 5000

Browse Source
This commit is contained in:
Daniel Povey 2022-07-28 14:45:45 +08:00
parent 0d038a6ea4
commit 633cbd551a
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_stateless7/train.py
View File

@ -929,7 +929,7 @@ def run(rank, world_size, args):
optimizer = PrAdam(model.parameters(), optimizer = PrAdam(model.parameters(),
lr=params.initial_lr, lr=params.initial_lr,
max_block_size=512, max_block_size=512,
lr_update_period=(200, 4000)) lr_update_period=(400, 5000))
scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs) scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)