mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-09 14:05:33 +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

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

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