Trait tch::nn::OptimizerConfig

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

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

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

Required Methods

Provided Methods

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

Implementors