{ }
Update '*var' according to the momentum scheme.
tf.raw_ops.ResourceApplyKerasMomentum(
var,
accum,
lr,
grad,
momentum,
use_locking=False,
use_nesterov=False,
name=None
)
Set use_nesterov = True if you want to use Nesterov momentum.
accum = accum * momentum - lr * grad var += accum
Returns | |
---|---|
The created Operation. |