refine lars (#36409)

eb722e34 · Zeng Jinle · GitHub · 8ffcc7c8 · eb722e34
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/optimizer.py python/paddle/fluid/optimizer.py +1 -1

未找到文件。
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -2086,7 +2086,7 @@ class LarsMomentumOptimizer(Optimizer):

        # create the momentum optimize op
        momentum_op = block.append_op(
-            type=self.type,
+            type=self.type if _lars_weight_decay != 0.0 else 'momentum',
            inputs=inputs,
            outputs=outputs,
            attrs=attrs,