未验证 提交 398b96c6 编写于 作者: S SmirnovKol 提交者: GitHub

Update optimizer.py (#43201)

上级 39903f72
...@@ -338,9 +338,6 @@ class Optimizer(object): ...@@ -338,9 +338,6 @@ class Optimizer(object):
adam.set_state_dict(opti_state_dict) adam.set_state_dict(opti_state_dict)
''' '''
if isinstance(self._learning_rate, LRScheduler):
self._learning_rate.set_dict(state_dict["LR_Scheduler"])
if isinstance(self._learning_rate, LRScheduler): if isinstance(self._learning_rate, LRScheduler):
self._learning_rate.set_state_dict(state_dict["LR_Scheduler"]) self._learning_rate.set_state_dict(state_dict["LR_Scheduler"])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册