[−][src]Trait tch::nn::OptimizerConfig

pub trait OptimizerConfig where
    Self: Sized, {
    fn build_copt(&self, lr: f64) -> Fallible<COptimizer>;

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

Required methods

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

Loading content...

Provided methods

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

Loading content...

Implementors

impl OptimizerConfig for Adam[src]

fn build(self, vs: &VarStore, lr: f64) -> Fallible<Optimizer<Self>>[src]

impl OptimizerConfig for RmsProp[src]

fn build(self, vs: &VarStore, lr: f64) -> Fallible<Optimizer<Self>>[src]

impl OptimizerConfig for Sgd[src]

fn build(self, vs: &VarStore, lr: f64) -> Fallible<Optimizer<Self>>[src]

Loading content...