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

release/1.7
Aurelius84 5 years ago committed by GitHub
parent 835ba4ec81
commit 6b4c33ee30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save