Skip to content

Commit 7d9bf24

Browse files
author
Feiyu Chan
authored
exclude lr scheduler's state from accumulators_holder (#33984)
* exclude lr scheduler's state from accumulators_holder * fix when there is no learning rate scheduler * make a copy of the loaded state dict to avoid modifying it
1 parent 46879ff commit 7d9bf24

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/paddle/optimizer/optimizer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,11 @@ def set_state_dict(self, state_dict):
286286
if isinstance(self._learning_rate, LRScheduler):
287287
self._learning_rate.set_state_dict(state_dict["LR_Scheduler"])
288288

289+
# NOTE: exclude learning rate scheduler's state from
290+
# _accumulators_holder.
291+
state_dict = state_dict.copy()
292+
if "LR_Scheduler" in state_dict:
293+
state_dict.pop("LR_Scheduler")
289294
self._accumulators_holder = state_dict
290295
for k, v in self._accumulators.items():
291296
for para_name, var_tmp in v.items():

0 commit comments

Comments
 (0)