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

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

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.