optimize memory allocation of optimizer (#53097)

b8f05ff2 · Tian · GitHub · b2d5cfab · b8f05ff2
隐藏空白更改
内联并排

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