未验证 提交 eb722e34 编写于 作者: Z Zeng Jinle 提交者: GitHub

refine lars (#36409)

上级 8ffcc7c8
...@@ -2086,7 +2086,7 @@ class LarsMomentumOptimizer(Optimizer): ...@@ -2086,7 +2086,7 @@ class LarsMomentumOptimizer(Optimizer):
# create the momentum optimize op # create the momentum optimize op
momentum_op = block.append_op( momentum_op = block.append_op(
type=self.type, type=self.type if _lars_weight_decay != 0.0 else 'momentum',
inputs=inputs, inputs=inputs,
outputs=outputs, outputs=outputs,
attrs=attrs, attrs=attrs,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册