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