Safe Haskell	None
Language	Haskell2010

Torch.Typed.Optim.CppOptim

Documentation

class CppOptimizer option where Source #

Minimal complete definition

initOptimizer

Methods

initOptimizer :: forall model (tensors :: [Type]). (Parameterized model, HMap' ToDependent (Parameters model) tensors, Castable (HList tensors) [ATenTensor]) => option -> model -> IO (CppOptimizerState option (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 option (Parameters model) -> Tensor dev dtype lossShape -> IO (model, CppOptimizerState option (Parameters model)) Source #

Instances

Instances details

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

runStep :: forall option model (tensors :: [Type]) (dev :: (DeviceType, Nat)) (dtype :: DType). (CppOptimizer option, Parameterized model, HMap' ToDependent (Parameters model) tensors, HMap' ToParameter tensors (Parameters model), Castable (HList tensors) [ATenTensor]) => model -> CppOptimizerState option (Parameters model) -> Loss dev dtype -> IO (model, CppOptimizerState option (Parameters model)) Source #

type CppOptimizerRef = ForeignPtr Optimizer Source #

data CppOptimizerState option (params :: [Type]) Source #

Constructors

CppOptimizerState option CppOptimizerRef

data ToParameter Source #

Constructors

ToParameter

Instances

Instances details

Apply' ToParameter (Tensor dev dtype shape) (Parameter dev dtype shape) Source #
Instance details Defined in Torch.Typed.Optim.CppOptim Methods apply' :: ToParameter -> Tensor dev dtype shape -> Parameter dev dtype shape Source #

data AdagradOptions Source #

Constructors

AdagradOptions
Fields adagradLr :: Double adagradLrDecay :: Double adagradWeightDecay :: Double adagradInitialAccumulatorValue :: Double adagradEps :: Double

Instances

Instances details

Default AdagradOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: AdagradOptions #
Show AdagradOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> AdagradOptions -> ShowS # show :: AdagradOptions -> String # showList :: [AdagradOptions] -> ShowS #
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 adamLr :: Double adamBetas :: (Double, Double) adamEps :: Double adamWeightDecay :: Double adamAmsgrad :: Bool

Instances

Instances details

Default AdamOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: AdamOptions #
Show AdamOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> AdamOptions -> ShowS # show :: AdamOptions -> String # showList :: [AdamOptions] -> ShowS #
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, 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 adamwLr :: Double adamwBetas :: (Double, Double) adamwEps :: Double adamwWeightDecay :: Double adamwAmsgrad :: Bool

Instances

Instances details

Default AdamwOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: AdamwOptions #
Show AdamwOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> AdamwOptions -> ShowS # show :: AdamwOptions -> String # showList :: [AdamwOptions] -> ShowS #
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, 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 lbfgsLr :: Double lbfgsMaxIter :: Int lbfgsMaxEval :: Int lbfgsToleranceGrad :: Double lbfgsToleranceChange :: Double lbfgsHistorySize :: Int lbfgsLineSearchFn :: Maybe String

Instances

Instances details

Default LbfgsOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: LbfgsOptions #
Show LbfgsOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> LbfgsOptions -> ShowS # show :: LbfgsOptions -> String # showList :: [LbfgsOptions] -> ShowS #
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, 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 rmspropLr :: Double rmspropAlpha :: Double rmspropEps :: Double rmspropWeightDecay :: Double rmspropMomentum :: Double rmspropCentered :: Bool

Instances

Instances details

Default RmspropOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: RmspropOptions #
Show RmspropOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> RmspropOptions -> ShowS # show :: RmspropOptions -> String # showList :: [RmspropOptions] -> ShowS #
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, 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 sgdLr :: Double sgdMomentum :: Double sgdDampening :: Double sgdWeightDecay :: Double sgdNesterov :: Bool

Instances

Instances details

Default SGDOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods def :: SGDOptions #
Show SGDOptions Source #
Instance details Defined in Torch.Optim.CppOptim Methods showsPrec :: Int -> SGDOptions -> ShowS # show :: SGDOptions -> String # showList :: [SGDOptions] -> ShowS #
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, 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 #