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

Remove optimizer.reset() which is not supported for Eve

Browse Source
This commit is contained in:
Daniel Povey 2022-05-16 12:13:36 +08:00
parent d58debdb3f
commit 9859e33c06
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

@ -704,7 +704,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.orthogonalize() mmodel.encoder.orthogonalize()
optimizer.reset() #optimizer.reset()
params.batch_idx_train += 1 params.batch_idx_train += 1
batch_size = len(batch["supervisions"]["text"]) batch_size = len(batch["supervisions"]["text"])