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.