Trait tch::nn::OptimizerConfig

source · 
pub trait OptimizerConfig
where
    Self: Sized,
{
    // Required method
    fn build_copt(&self, lr: f64) -> Result<COptimizer, TchError>;

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

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

Required Methods§

source

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

Provided Methods§

source

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

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

Object Safety§

This trait is not object safe.

Implementors§