fix l1 decay for inplace (#32717)

PaddlePaddle · May 6, 2021 · efdb0a7 · efdb0a7
1 parent c5ae21f
commit efdb0a7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/python/paddle/fluid/regularizer.py b/python/paddle/fluid/regularizer.py
@@ -326,19 +326,21 @@ def __call__(self, param, grad, block):
         assert isinstance(block, framework.Block)
 
         if framework.in_dygraph_mode():
+            sign = block.create_var(dtype=param.dtype, shape=param.shape)
             decay = block.create_var(dtype=param.dtype, shape=param.shape)
         else:
+            sign = block.create_var(
+                dtype=param.dtype, shape=param.shape, lod_level=param.lod_level)
             decay = block.create_var(
                 dtype=param.dtype, shape=param.shape, lod_level=param.lod_level)
 
         # Append sign op
-        block.append_op(
-            type='sign', inputs={"X": param}, outputs={"Out": decay})
+        block.append_op(type='sign', inputs={"X": param}, outputs={"Out": sign})
 
         # Append scale op to the output of sign op
         block.append_op(
             type='scale',
-            inputs={"X": decay},
+            inputs={"X": sign},
             outputs={"Out": decay},
             attrs={"scale": self._regularization_coeff})