Trait tch::nn::OptimizerConfig [−][src]
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.