{ }
Update '*var' according to the momentum scheme.
tf.raw_ops.ApplyMomentum(
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 + grad var -= lr * accum
Returns | |
---|---|
A mutable Tensor . Has the same type as var .
|