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

Fix an error in TDNN-LSTM training.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-07-31 15:55:42 +08:00
parent 398ed80d7a
commit c9222bdb09
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/tdnn_lstm_ctc/train.py
View File

@ -501,6 +501,7 @@ def run(rank, world_size, args):
) )
scheduler = StepLR(optimizer, step_size=8, gamma=0.1) scheduler = StepLR(optimizer, step_size=8, gamma=0.1)
if checkpoints:
optimizer.load_state_dict(checkpoints["optimizer"]) optimizer.load_state_dict(checkpoints["optimizer"])
scheduler.load_state_dict(checkpoints["scheduler"]) scheduler.load_state_dict(checkpoints["scheduler"])