From 9859e33c06eeed1c307d7d91e24e15750ee61307 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 16 May 2022 12:13:36 +0800 Subject: [PATCH] Remove optimizer.reset() which is not supported for Eve --- egs/librispeech/ASR/pruned_transducer_stateless4b/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py b/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py index 83e7bf0d6..53ebca8ac 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py @@ -704,7 +704,7 @@ def train_one_epoch( if params.batch_idx_train % 2000 == 0 and params.batch_idx_train > 0: mmodel = model.module if hasattr(model, 'module') else model mmodel.encoder.orthogonalize() - optimizer.reset() + #optimizer.reset() params.batch_idx_train += 1 batch_size = len(batch["supervisions"]["text"])