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

fix train.py

Browse Source
This commit is contained in:
Daniel Povey 2022-05-16 19:41:16 +08:00
parent d61c8aa3bc
commit 2f2934a115
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_stateless4b/train.py
View File

@ -703,7 +703,7 @@ def train_one_epoch(
if params.batch_idx_train % 2000 == 0 and params.batch_idx_train > 0: if params.batch_idx_train % 2000 == 0 and params.batch_idx_train > 0:
mmodel = model.module if hasattr(model, 'module') else model mmodel = model.module if hasattr(model, 'module') else model
mmodel.encoder.diagonalize() mmodel.diagonalize()
#optimizer.reset() #optimizer.reset()
params.batch_idx_train += 1 params.batch_idx_train += 1