Struct tch::nn::Sgd

pub struct Sgd {
    pub momentum: f64,
    pub dampening: f64,
    pub wd: f64,
    pub nesterov: bool,
}

Parameters for the SGD optimizer.

Fields

momentum: f64

dampening: f64

wd: f64

nesterov: bool

Trait Implementations

impl Clone for Sgd

fn clone(&self) -> Sgd

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Sgd

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for Sgd

fn default() -> Self

Returns the “default value” for a type.

impl OptimizerConfig for Sgd

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

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

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

impl Copy for Sgd