Safe Haskell	None
Language	Haskell2010

Torch.Optim.CppOptim

Documentation

type CppOptimizerRef = ForeignPtr Optimizer Source #

data CppOptimizerState option Source #

Constructors

CppOptimizerState option CppOptimizerRef

Instances

Instances details

CppOptimizer option => Optimizer (CppOptimizerState option) Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

step :: LearningRate -> Gradients -> [Tensor] -> CppOptimizerState option -> ([Tensor], CppOptimizerState option) Source #

runStep :: Parameterized model => model -> CppOptimizerState option -> Loss -> LearningRate -> IO (model, CppOptimizerState option) Source #

runStep' :: Parameterized model => model -> CppOptimizerState option -> Gradients -> LearningRate -> IO (model, CppOptimizerState option) Source #

stepWithGenerator :: CppOptimizerState option -> ForeignPtr Generator -> ([Tensor] -> ForeignPtr Generator -> IO (Tensor, ForeignPtr Generator)) -> IO (Tensor, ForeignPtr Generator) Source #

class CppOptimizer option where Source #

Minimal complete definition

initOptimizer

Methods

initOptimizer :: Parameterized model => option -> model -> IO (CppOptimizerState option) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState option -> Tensor -> IO (model, CppOptimizerState option) Source #

Instances

Instances details

CppOptimizer AdagradOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdagradOptions -> model -> IO (CppOptimizerState AdagradOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdagradOptions -> Tensor -> IO (model, CppOptimizerState AdagradOptions) Source #

CppOptimizer AdamOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdamOptions -> model -> IO (CppOptimizerState AdamOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdamOptions -> Tensor -> IO (model, CppOptimizerState AdamOptions) Source #

CppOptimizer AdamwOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdamwOptions -> model -> IO (CppOptimizerState AdamwOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdamwOptions -> Tensor -> IO (model, CppOptimizerState AdamwOptions) Source #

CppOptimizer LbfgsOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => LbfgsOptions -> model -> IO (CppOptimizerState LbfgsOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState LbfgsOptions -> Tensor -> IO (model, CppOptimizerState LbfgsOptions) Source #

CppOptimizer RmspropOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => RmspropOptions -> model -> IO (CppOptimizerState RmspropOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState RmspropOptions -> Tensor -> IO (model, CppOptimizerState RmspropOptions) Source #

CppOptimizer SGDOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => SGDOptions -> model -> IO (CppOptimizerState SGDOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState SGDOptions -> Tensor -> IO (model, CppOptimizerState SGDOptions) Source #

data AdagradOptions Source #

Constructors

AdagradOptions

Fields

adagradLr :: Double
adagradLrDecay :: Double
adagradWeightDecay :: Double
adagradInitialAccumulatorValue :: Double
adagradEps :: Double

Instances

Instances details

Show AdagradOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdagradOptions -> ShowS #

show :: AdagradOptions -> String #

showList :: [AdagradOptions] -> ShowS #

Default AdagradOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdagradOptions #

Eq AdagradOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: AdagradOptions -> AdagradOptions -> Bool #

(/=) :: AdagradOptions -> AdagradOptions -> Bool #

CppOptimizer AdagradOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdagradOptions -> model -> IO (CppOptimizerState AdagradOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdagradOptions -> Tensor -> IO (model, CppOptimizerState AdagradOptions) Source #

data AdamOptions Source #

Constructors

AdamOptions

Fields

Instances

Instances details

Show AdamOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdamOptions -> ShowS #

show :: AdamOptions -> String #

showList :: [AdamOptions] -> ShowS #

Default AdamOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdamOptions #

Eq AdamOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: AdamOptions -> AdamOptions -> Bool #

(/=) :: AdamOptions -> AdamOptions -> Bool #

CppOptimizer AdamOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdamOptions -> model -> IO (CppOptimizerState AdamOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdamOptions -> Tensor -> IO (model, CppOptimizerState AdamOptions) Source #

CppOptimizer AdamOptions Source #

Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, HList tensors) [ATenTensor]) => AdamOptions -> model -> IO (CppOptimizerState AdamOptions (Parameters model)) Source #

unsafeStep :: forall model (dev :: (DeviceType, Nat)) (dtype :: DType) (lossShape :: [Nat]) (tensors :: [Type]) res. (Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState AdamOptions (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState AdamOptions (Parameters model)) Source #

data AdamwOptions Source #

Constructors

AdamwOptions

Fields

adamwLr :: Double
adamwBetas :: (Double, Double)
adamwEps :: Double
adamwWeightDecay :: Double
adamwAmsgrad :: Bool

Instances

Instances details

Show AdamwOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdamwOptions -> ShowS #

show :: AdamwOptions -> String #

showList :: [AdamwOptions] -> ShowS #

Default AdamwOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdamwOptions #

Eq AdamwOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: AdamwOptions -> AdamwOptions -> Bool #

(/=) :: AdamwOptions -> AdamwOptions -> Bool #

CppOptimizer AdamwOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => AdamwOptions -> model -> IO (CppOptimizerState AdamwOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState AdamwOptions -> Tensor -> IO (model, CppOptimizerState AdamwOptions) Source #

CppOptimizer AdamwOptions Source #

Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, HList tensors) [ATenTensor]) => AdamwOptions -> model -> IO (CppOptimizerState AdamwOptions (Parameters model)) Source #

unsafeStep :: forall model (dev :: (DeviceType, Nat)) (dtype :: DType) (lossShape :: [Nat]) (tensors :: [Type]) res. (Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState AdamwOptions (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState AdamwOptions (Parameters model)) Source #

data LbfgsOptions Source #

Constructors

LbfgsOptions

Fields

lbfgsLr :: Double
lbfgsMaxIter :: Int
lbfgsMaxEval :: Int
lbfgsToleranceGrad :: Double
lbfgsToleranceChange :: Double
lbfgsHistorySize :: Int
lbfgsLineSearchFn :: Maybe String

Instances

Instances details

Show LbfgsOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> LbfgsOptions -> ShowS #

show :: LbfgsOptions -> String #

showList :: [LbfgsOptions] -> ShowS #

Default LbfgsOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: LbfgsOptions #

Eq LbfgsOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: LbfgsOptions -> LbfgsOptions -> Bool #

(/=) :: LbfgsOptions -> LbfgsOptions -> Bool #

CppOptimizer LbfgsOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => LbfgsOptions -> model -> IO (CppOptimizerState LbfgsOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState LbfgsOptions -> Tensor -> IO (model, CppOptimizerState LbfgsOptions) Source #

CppOptimizer LbfgsOptions Source #

Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, HList tensors) [ATenTensor]) => LbfgsOptions -> model -> IO (CppOptimizerState LbfgsOptions (Parameters model)) Source #

unsafeStep :: forall model (dev :: (DeviceType, Nat)) (dtype :: DType) (lossShape :: [Nat]) (tensors :: [Type]) res. (Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState LbfgsOptions (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState LbfgsOptions (Parameters model)) Source #

data RmspropOptions Source #

Constructors

RmspropOptions

Fields

rmspropLr :: Double
rmspropAlpha :: Double
rmspropEps :: Double
rmspropWeightDecay :: Double
rmspropMomentum :: Double
rmspropCentered :: Bool

Instances

Instances details

Show RmspropOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> RmspropOptions -> ShowS #

show :: RmspropOptions -> String #

showList :: [RmspropOptions] -> ShowS #

Default RmspropOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: RmspropOptions #

Eq RmspropOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: RmspropOptions -> RmspropOptions -> Bool #

(/=) :: RmspropOptions -> RmspropOptions -> Bool #

CppOptimizer RmspropOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => RmspropOptions -> model -> IO (CppOptimizerState RmspropOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState RmspropOptions -> Tensor -> IO (model, CppOptimizerState RmspropOptions) Source #

CppOptimizer RmspropOptions Source #

Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, HList tensors) [ATenTensor]) => RmspropOptions -> model -> IO (CppOptimizerState RmspropOptions (Parameters model)) Source #

unsafeStep :: forall model (dev :: (DeviceType, Nat)) (dtype :: DType) (lossShape :: [Nat]) (tensors :: [Type]) res. (Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState RmspropOptions (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState RmspropOptions (Parameters model)) Source #

data SGDOptions Source #

Constructors

SGDOptions

Fields

sgdLr :: Double
sgdMomentum :: Double
sgdDampening :: Double
sgdWeightDecay :: Double
sgdNesterov :: Bool

Instances

Instances details

Show SGDOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> SGDOptions -> ShowS #

show :: SGDOptions -> String #

showList :: [SGDOptions] -> ShowS #

Default SGDOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: SGDOptions #

Eq SGDOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

(==) :: SGDOptions -> SGDOptions -> Bool #

(/=) :: SGDOptions -> SGDOptions -> Bool #

CppOptimizer SGDOptions Source #

Instance details

Defined in Torch.Optim.CppOptim

Methods

initOptimizer :: Parameterized model => SGDOptions -> model -> IO (CppOptimizerState SGDOptions) Source #

unsafeStep :: Parameterized model => model -> CppOptimizerState SGDOptions -> Tensor -> IO (model, CppOptimizerState SGDOptions) Source #

CppOptimizer SGDOptions Source #

Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, HList tensors) [ATenTensor]) => SGDOptions -> model -> IO (CppOptimizerState SGDOptions (Parameters model)) Source #

unsafeStep :: forall model (dev :: (DeviceType, Nat)) (dtype :: DType) (lossShape :: [Nat]) (tensors :: [Type]) res. (Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState SGDOptions (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState SGDOptions (Parameters model)) Source #

saveState :: CppOptimizerState option -> FilePath -> IO () Source #

loadState :: CppOptimizerState option -> FilePath -> IO () Source #