public static class ApplyMomentum.Options extends Object
ApplyMomentum
Modifier and Type | Method and Description |
---|---|
ApplyMomentum.Options |
useLocking(Boolean useLocking) |
ApplyMomentum.Options |
useNesterov(Boolean useNesterov) |
public ApplyMomentum.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 ApplyMomentum.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.