remove additional warnning in layer.to

PaddlePaddle · Oct 25, 2021 · 3fd7e25 · 3fd7e25
1 parent 4f12d7e
commit 3fd7e25
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/python/paddle/fluid/dygraph/layers.py b/python/paddle/fluid/dygraph/layers.py
@@ -1479,17 +1479,17 @@ def _apply(self, func, device, dtype, blocking):
                         param_applied.is_distributed = param.is_distributed
                     self._parameters[key] = param_applied
 
-                    if param.grad is not None:
-                        with no_grad():
-                            grad_applied = func(param._grad_ivar(), device,
-                                                dtype, blocking)
-
-                            grad_applied.stop_gradient = param._grad_ivar(
-                            ).stop_gradient
-                            if hasattr(param._grad_ivar(), 'is_distributed'):
-                                grad_applied.is_distributed = param._grad_ivar(
-                                ).is_distributed
-                            self._parameters[key]._set_grad_ivar(grad_applied)
+                if param.grad is not None:
+                    with no_grad():
+                        grad_applied = func(param._grad_ivar(), device, dtype,
+                                            blocking)
+
+                        grad_applied.stop_gradient = param._grad_ivar(
+                        ).stop_gradient
+                        if hasattr(param._grad_ivar(), 'is_distributed'):
+                            grad_applied.is_distributed = param._grad_ivar(
+                            ).is_distributed
+                        self._parameters[key]._set_grad_ivar(grad_applied)
 
             self._parameters_transform_map[id(param)] = [param_applied, key]