Struct RmsPropState 

pub struct RmsPropState<B: Backend, const D: usize> {
    pub square_avg: SquareAvgState<B, D>,
    pub centered: CenteredState<B, D>,
    pub momentum: Option<RmsPropMomentumState<B, D>>,
}

State of RmsProp

Fields

square_avg: SquareAvgState<B, D>

Current squared average state.

centered: CenteredState<B, D>

Current centered state

momentum: Option<RmsPropMomentumState<B, D>>

Current gradient momentum, if any.

Implementations

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

pub fn new( square_avg: SquareAvgState<B, D>, centered: CenteredState<B, D>, momentum: Option<RmsPropMomentumState<B, D>>, ) -> Self

Constructs a new RmsPropState.

Trait Implementations

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

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

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

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

Type of the item that can be serialized and deserialized.

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>, device: &B::Device, ) -> Self

Convert the given item into a record.