Struct tch::nn::Adam

pub struct Adam {
    pub beta1: f64,
    pub beta2: f64,
    pub wd: f64,
    pub eps: f64,
    pub amsgrad: bool,
}

Parameters for the Adam optimizer.

Fields

beta1: f64

beta2: f64

wd: f64

eps: f64

amsgrad: bool

Implementations

impl Adam

pub fn beta1(self, b: f64) -> Self

pub fn beta2(self, b: f64) -> Self

pub fn wd(self, w: f64) -> Self

pub fn eps(self, e: f64) -> Self

pub fn amsgrad(self, a: bool) -> Self

Trait Implementations

impl Clone for Adam

fn clone(&self) -> Adam

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Adam

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

Formats the value using the given formatter.

impl Default for Adam

fn default() -> Self

Returns the “default value” for a type.

impl OptimizerConfig for Adam

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 Adam