optimize memory allocation of optimizer (#53138)

27159c42 · Tian · GitHub · 4e939c89 · 27159c42
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/optimizer/optimizer.py python/paddle/optimizer/optimizer.py +1 -1

未找到文件。
--- a/python/paddle/optimizer/optimizer.py
+++ b/python/paddle/optimizer/optimizer.py
@@ -774,7 +774,7 @@ class Optimizer:
                ), "Optimizer set error, {} should in state dict".format(
                    var_name
                )
-                var.set_value(self._accumulators_holder[var_name])
+                var.set_value(self._accumulators_holder.pop(var_name))

        self._accumulators[name][param.name] = var
        return var