Package | Description |
---|---|
org.bytedeco.pytorch |
Modifier and Type | Method and Description |
---|---|
Tensor |
LBFGS.step(LossClosure closure) |
Tensor |
AdamW.step(LossClosure closure) |
Tensor |
Optimizer.step(LossClosure closure)
A loss function closure, which is expected to return the loss value.
|
Tensor |
SGD.step(LossClosure closure) |
Tensor |
Adagrad.step(LossClosure closure) |
Tensor |
RMSprop.step(LossClosure closure) |
Tensor |
Adam.step(LossClosure closure) |
Copyright © 2024. All rights reserved.