ResourceApplyMomentum.Options

public static class ResourceApplyMomentum.Options

Optional attributes for ResourceApplyMomentum

Public Methods

ResourceApplyMomentum.Options
useLocking (Boolean useLocking)
ResourceApplyMomentum.Options
useNesterov (Boolean useNesterov)

Inherited Methods

Public Methods

public ResourceApplyMomentum.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 ResourceApplyMomentum.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.