Safe Haskell	Safe-Inferred
Language	Haskell2010

Torch.Typed.Optim.CppOptim

Documentation

class CppOptimizer option where Source #

Minimal complete definition

initOptimizer

Methods

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

unsafeStep :: forall model dev dtype lossShape tensors 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 #

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 CppOptimizerState option (params :: [*]) Source #

Constructors

CppOptimizerState option CppOptimizerRef

type CppOptimizerRef = ForeignPtr Optimizer Source #

runStep :: (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 #

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 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 adamLr :: Double adamBetas :: (Double, Double) adamEps :: Double adamWeightDecay :: Double adamAmsgrad :: Bool

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