fix merged_momentum regular bug

PaddlePaddle · Dec 22, 2021 · 151c2ef · 151c2ef
1 parent fabc058
commit 151c2ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/optimizer/momentum.py b/python/paddle/optimizer/momentum.py
@@ -380,7 +380,7 @@ def _multi_tensor_init(self, target_block, parameters):
                 if isinstance(param.regularizer, L2DecayRegularizer):
                     regularization_method = "l2_decay"
                     regularization_coeff = param.regularizer._regularization_coeff
-                else:
+                elif param.regularizer is not None:
                     regularization_method = ""
                     regularization_coeff = 0.0
             if param.dtype == paddle.float32: