tch_plus::nn

Trait 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.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§