Class ApplyMomentum.Options

java.lang.Object

org.tensorflow.op.train.ApplyMomentum.Options

Enclosing class:

ApplyMomentum<T extends TType>

public static class ApplyMomentum.Options
extends Object

Optional attributes for ApplyMomentum

Method Summary

Modifier and Type	Method	Description
ApplyMomentum.Options	useLocking(Boolean useLocking)	Sets the useLocking option.
ApplyMomentum.Options	useNesterov(Boolean useNesterov)	Sets the useNesterov option.

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

useLocking

public ApplyMomentum.Options useLocking(Boolean useLocking)

Sets the useLocking option.

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.

Returns:

this Options instance.

useNesterov

public ApplyMomentum.Options useNesterov(Boolean useNesterov)

Sets the useNesterov option.

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.

Returns:

this Options instance.