Trait tch::nn::OptimizerConfig

source · 
pub trait OptimizerConfigwhere
    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.

Implementors§