Trait tch::nn::OptimizerConfig

pub trait OptimizerConfigwhere
    Self: Sized,{
    fn build_copt(&self, lr: f64) -> Result<COptimizer, TchError>;

    fn build(self, vs: &VarStore, lr: f64) -> Result<Optimizer, TchError> { ... }
}

Optimizer configurations. These configs can be used to build optimizer.

Required Methods

fn build_copt(&self, lr: f64) -> Result<COptimizer, TchError>

Provided Methods

fn build(self, vs: &VarStore, lr: f64) -> Result<Optimizer, TchError>

Builds an optimizer with the specified learning rate handling variables stored in vs.

Implementors

impl OptimizerConfig for Adam

impl OptimizerConfig for AdamW

impl OptimizerConfig for RmsProp

impl OptimizerConfig for Sgd