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

refine lars (#36409)

上级 8ffcc7c8
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册