From 1d5e509261b7434c016a1425a0674885d49de0e9 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 14 Sep 2021 13:13:48 +0800 Subject: [PATCH] Fix to madam.py, RE optimizer state --- egs/librispeech/ASR/conformer_ctc_bn/madam.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/egs/librispeech/ASR/conformer_ctc_bn/madam.py b/egs/librispeech/ASR/conformer_ctc_bn/madam.py index 1cf0d322f..0ee0655d4 100644 --- a/egs/librispeech/ASR/conformer_ctc_bn/madam.py +++ b/egs/librispeech/ASR/conformer_ctc_bn/madam.py @@ -929,6 +929,7 @@ class Foam(object): """Return state_dict.""" return { "_step": self._step, + "optimizer": self.optimizer.state_dict(), } def load_state_dict(self, state_dict): @@ -1049,6 +1050,7 @@ class Gloam(object): def state_dict(self): """Return state_dict.""" return { + "optimizer": self.optimizer.state_dict(), "_step": self._step, "_epoch": self._epoch, }