未验证 提交 6b4c33ee 编写于 作者: A Aurelius84 提交者: GitHub

fix decay param in DecayAdagrad test=develop (#22026)

上级 835ba4ec
...@@ -2098,7 +2098,8 @@ class DecayedAdagradOptimizer(Optimizer): ...@@ -2098,7 +2098,8 @@ class DecayedAdagradOptimizer(Optimizer):
}, },
outputs={"ParamOut": param_and_grad[0], outputs={"ParamOut": param_and_grad[0],
"MomentOut": moment_acc}, "MomentOut": moment_acc},
attrs={"epsilon": self._epsilon}, attrs={"epsilon": self._epsilon,
"decay": self._decay},
stop_gradient=True) stop_gradient=True)
return decayed_adagrad_op return decayed_adagrad_op
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册