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

Avoid using lr from checkpoint. (#1781)

Browse Source
This commit is contained in:
Fangjun Kuang 2024-10-28 00:59:04 +08:00 committed by GitHub
parent 37a1420603
commit 05f756390c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/librispeech/ASR/zipformer/optim.py
View File

@ -787,7 +787,9 @@ class LRScheduler(object):
is not the optimizer. is not the optimizer.
""" """
return { return {
"base_lrs": self.base_lrs, # the user might try to override the base_lr, so don't include this in the state.
# previously they were included.
# "base_lrs": self.base_lrs,
"epoch": self.epoch, "epoch": self.epoch,
"batch": self.batch, "batch": self.batch,
} }
@ -799,7 +801,12 @@ class LRScheduler(object):
state_dict (dict): scheduler state. Should be an object returned state_dict (dict): scheduler state. Should be an object returned
from a call to :meth:`state_dict`. from a call to :meth:`state_dict`.
""" """
# the things with base_lrs are a work-around for a previous problem
# where base_lrs were written with the state dict.
base_lrs = self.base_lrs
self.__dict__.update(state_dict) self.__dict__.update(state_dict)
self.base_lrs = base_lrs
def get_last_lr(self) -> List[float]: def get_last_lr(self) -> List[float]:
"""Return last computed learning rate by current scheduler. Will be a list of float.""" """Return last computed learning rate by current scheduler. Will be a list of float."""