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

Small code beautification

Browse Source
This commit is contained in:
Daniel Povey 2021-09-23 11:39:25 +08:00
parent 2213457bd3
commit 3415dab779
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc_bn_2d/madam.py
View File

@ -806,7 +806,7 @@ class Moam(object):
"""Load state_dict.""" """Load state_dict."""
for key, value in state_dict.items(): for key, value in state_dict.items():
if key == "optimizer": if key == "optimizer":
self.optimizer.load_state_dict(state_dict["optimizer"]) self.optimizer.load_state_dict(value)
else: else:
setattr(self, key, value) setattr(self, key, value)
@ -1059,7 +1059,7 @@ class Gloam(object):
"""Load state_dict. This is compatible with reading a Moam state_dict""" """Load state_dict. This is compatible with reading a Moam state_dict"""
for key, value in state_dict.items(): for key, value in state_dict.items():
if key == "optimizer": if key == "optimizer":
self.optimizer.load_state_dict(state_dict["optimizer"]) self.optimizer.load_state_dict(value)
elif key == '_step': elif key == '_step':
self._step = value self._step = value
elif key == '_epoch': elif key == '_epoch':