Torch.Typed.Optim.CppOptim

Documentation

class CppOptimizer option where

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

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

Instances

CppOptimizer AdamOptions

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

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

CppOptimizer AdamwOptions

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

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

CppOptimizer LbfgsOptions

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

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

CppOptimizer RmspropOptions

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

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

CppOptimizer SGDOptions

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

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

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

type CppOptimizerRef = ForeignPtr Optimizer

data CppOptimizerState option (params :: [Type])

Constructors

CppOptimizerState option CppOptimizerRef

data ToParameter

Constructors

ToParameter

Instances

Apply' ToParameter (Tensor dev dtype shape) (Parameter dev dtype shape)

Defined in Torch.Typed.Optim.CppOptim

Methods

apply' :: ToParameter -> Tensor dev dtype shape -> Parameter dev dtype shape

data AdagradOptions

Constructors

AdagradOptions

Fields

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

Instances

Show AdagradOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdagradOptions -> ShowS

show :: AdagradOptions -> String

showList :: [AdagradOptions] -> ShowS

Default AdagradOptions

Defined in Torch.Optim.CppOptim

Methods

def :: AdagradOptions

Eq AdagradOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer AdagradOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

data AdamOptions

Constructors

AdamOptions

Fields

• adamLr :: Double
• adamBetas :: (Double, Double)
• adamEps :: Double
• adamWeightDecay :: Double
• adamAmsgrad :: Bool

Instances

Show AdamOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdamOptions -> ShowS

show :: AdamOptions -> String

showList :: [AdamOptions] -> ShowS

Default AdamOptions

Defined in Torch.Optim.CppOptim

Methods

def :: AdamOptions

Eq AdamOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer AdamOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer AdamOptions

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

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

data AdamwOptions

Constructors

AdamwOptions

Fields

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

Instances

Show AdamwOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> AdamwOptions -> ShowS

show :: AdamwOptions -> String

showList :: [AdamwOptions] -> ShowS

Default AdamwOptions

Defined in Torch.Optim.CppOptim

Methods

def :: AdamwOptions

Eq AdamwOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer AdamwOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer AdamwOptions

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

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

data LbfgsOptions

Constructors

LbfgsOptions

Fields

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

Instances

Show LbfgsOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> LbfgsOptions -> ShowS

show :: LbfgsOptions -> String

showList :: [LbfgsOptions] -> ShowS

Default LbfgsOptions

Defined in Torch.Optim.CppOptim

Methods

def :: LbfgsOptions

Eq LbfgsOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer LbfgsOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer LbfgsOptions

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

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

data RmspropOptions

Constructors

RmspropOptions

Fields

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

Instances

Show RmspropOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> RmspropOptions -> ShowS

show :: RmspropOptions -> String

showList :: [RmspropOptions] -> ShowS

Default RmspropOptions

Defined in Torch.Optim.CppOptim

Methods

def :: RmspropOptions

Eq RmspropOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer RmspropOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer RmspropOptions

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

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

data SGDOptions

Constructors

SGDOptions

Fields

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

Instances

Show SGDOptions

Defined in Torch.Optim.CppOptim

Methods

showsPrec :: Int -> SGDOptions -> ShowS

show :: SGDOptions -> String

showList :: [SGDOptions] -> ShowS

Default SGDOptions

Defined in Torch.Optim.CppOptim

Methods

def :: SGDOptions

Eq SGDOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer SGDOptions

Defined in Torch.Optim.CppOptim

Methods

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

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

CppOptimizer SGDOptions

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

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