hasktorch-0.2.0.0: Functional differentiable programming in Haskell
Safe HaskellSafe-Inferred
LanguageHaskell2010

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

Instances

Instances details
Show AdagradOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdagradOptions -> ShowS Source #

show :: AdagradOptions -> String Source #

showList :: [AdagradOptions] -> ShowS Source #

Default AdagradOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdagradOptions Source #

Eq AdagradOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

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

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 Source #

show :: AdamOptions -> String Source #

showList :: [AdamOptions] -> ShowS Source #

Default AdamOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdamOptions Source #

Eq AdamOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

(/=) :: AdamOptions -> AdamOptions -> Bool 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 AdamOptions Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (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

Instances

Instances details
Show AdamwOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdamwOptions -> ShowS Source #

show :: AdamwOptions -> String Source #

showList :: [AdamwOptions] -> ShowS Source #

Default AdamwOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: AdamwOptions Source #

Eq AdamwOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

(/=) :: AdamwOptions -> AdamwOptions -> Bool 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 AdamwOptions Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (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

Instances

Instances details
Show LbfgsOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> LbfgsOptions -> ShowS Source #

show :: LbfgsOptions -> String Source #

showList :: [LbfgsOptions] -> ShowS Source #

Default LbfgsOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: LbfgsOptions Source #

Eq LbfgsOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

(/=) :: LbfgsOptions -> LbfgsOptions -> Bool 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 LbfgsOptions Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (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

Instances

Instances details
Show RmspropOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> RmspropOptions -> ShowS Source #

show :: RmspropOptions -> String Source #

showList :: [RmspropOptions] -> ShowS Source #

Default RmspropOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: RmspropOptions Source #

Eq RmspropOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

(/=) :: RmspropOptions -> RmspropOptions -> Bool 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 RmspropOptions Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (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

Instances

Instances details
Show SGDOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> SGDOptions -> ShowS Source #

show :: SGDOptions -> String Source #

showList :: [SGDOptions] -> ShowS Source #

Default SGDOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

def :: SGDOptions Source #

Eq SGDOptions Source # 
Instance details

Defined in Torch.Optim.CppOptim

Methods

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

(/=) :: SGDOptions -> SGDOptions -> Bool 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 #

CppOptimizer SGDOptions Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (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 #