public static class ResourceApplyMomentum.Options extends Object
ResourceApplyMomentum
Modifier and Type | Method and Description |
---|---|
ResourceApplyMomentum.Options |
useLocking(Boolean useLocking)
Sets the useLocking option.
|
ResourceApplyMomentum.Options |
useNesterov(Boolean useNesterov)
Sets the useNesterov option.
|
public ResourceApplyMomentum.Options useLocking(Boolean useLocking)
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)
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.Copyright © 2015–2022. All rights reserved.