Struct tch::nn::AdamW

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

Parameters for the AdamW optimizer.

Fields

beta1: f64

beta2: f64

wd: f64

eps: f64

amsgrad: bool

Implementations

impl AdamW

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 AdamW

fn clone(&self) -> AdamW

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for AdamW

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

Formats the value using the given formatter.

impl Default for AdamW

fn default() -> Self

Returns the “default value” for a type.

impl OptimizerConfig for AdamW

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 AdamW