Struct burn_core::optim::AdamState

pub struct AdamState<B: Backend, const D: usize> { /* private fields */ }

Implementations

impl<B: Backend, const D: usize> AdamState<B, D>

pub fn new( weight_decay: Option<WeightDecayState<B, D>>, momentum: AdaptiveMomentumState<B, D> ) -> Self

Constructs a new AdamState.

Trait Implementations

impl<B: Clone + Backend, const D: usize> Clone for AdamState<B, D>

fn clone(&self) -> AdamState<B, D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<B: Backend, const D: usize> Record for AdamState<B, D>

type Item<S: PrecisionSettings> = AdamStateItem<B, D, S>

fn into_item<S: PrecisionSettings>(self) -> Self::Item<S>

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S: PrecisionSettings>(item: Self::Item<S>) -> Self

Convert the given item into a record.