| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
Torch.Internal.Managed.Optim
Documentation
adagrad :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
rmsprop :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
sgd :: CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
adam :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
adamw :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
lbfgs :: CDouble -> CInt -> CInt -> CDouble -> CDouble -> CInt -> Maybe (ForeignPtr StdString) -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #
getParams :: ForeignPtr Optimizer -> IO (ForeignPtr TensorList) Source #
step :: ForeignPtr Optimizer -> (ForeignPtr TensorList -> IO (ForeignPtr Tensor)) -> IO (ForeignPtr Tensor) Source #
stepWithGenerator :: ForeignPtr Optimizer -> ForeignPtr Generator -> (ForeignPtr TensorList -> ForeignPtr Generator -> IO (ForeignPtr (StdTuple '(Tensor, Generator)))) -> IO (ForeignPtr (StdTuple '(Tensor, Generator))) Source #
unsafeStep :: ForeignPtr Optimizer -> ForeignPtr Tensor -> IO (ForeignPtr TensorList) Source #
save :: ForeignPtr Optimizer -> ForeignPtr StdString -> IO () Source #
load :: ForeignPtr Optimizer -> ForeignPtr StdString -> IO () Source #