fix bug

PaddlePaddle · Dec 22, 2021 · 6416a0e · 6416a0e · paddle-bot-old · Dec 22, 2021
1 parent 151c2ef
commit 6416a0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/optimizer/momentum.py b/python/paddle/optimizer/momentum.py
@@ -192,7 +192,7 @@ def __init__(self,
 
     def _update_regularization(self, weight_decay):
         reg_method = ""
-        reg_coeff = 0
+        reg_coeff = 0.0
 
         if (isinstance(weight_decay, L2DecayRegularizer)):
             reg_method = "l2_decay"
@@ -306,7 +306,7 @@ def _append_optimize_op(self, block, param_and_grad):
             # the param's regularization has been done before, we avoid do l2decay in momentum.
             elif param.regularizer is not None:
                 regularization_method = ""
-                regularization_coeff = 0
+                regularization_coeff = 0.0
 
         find_master = self._multi_precision and param_and_grad[
             0].dtype == core.VarDesc.VarType.FP16