ApplyMomentum.Options

public static class ApplyMomentum.Options

Optional attributes for ApplyMomentum

Public Methods

ApplyMomentum.Options
useLocking (Boolean useLocking)
ApplyMomentum.Options
useNesterov (Boolean useNesterov)

Inherited Methods

Public Methods

public ApplyMomentum.Options useLocking (Boolean useLocking)

Parameters
useLocking If `True`, updating of the var and accum tensors will be protected by a lock; otherwise the behavior is undefined, but may exhibit less contention.

public ApplyMomentum.Options useNesterov (Boolean useNesterov)

Parameters
useNesterov If `True`, the tensor passed to compute grad will be var - lr * momentum * accum, so in the end, the var you get is actually var - lr * momentum * accum.