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>>,
}Expand description
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§
Source§impl<B: Backend, const D: usize> RmsPropState<B, D>
impl<B: Backend, const D: usize> RmsPropState<B, D>
Sourcepub fn new(
square_avg: SquareAvgState<B, D>,
centered: CenteredState<B, D>,
momentum: Option<RmsPropMomentumState<B, D>>,
) -> Self
pub fn new( square_avg: SquareAvgState<B, D>, centered: CenteredState<B, D>, momentum: Option<RmsPropMomentumState<B, D>>, ) -> Self
Constructs a new RmsPropState.
Trait Implementations§
Source§impl<B: Backend, const D: usize> Record<B> for RmsPropState<B, D>
impl<B: Backend, const D: usize> Record<B> for RmsPropState<B, D>
Source§type Item<S: PrecisionSettings> = RmsPropStateItem<B, D, S>
type Item<S: PrecisionSettings> = RmsPropStateItem<B, D, S>
Type of the item that can be serialized and deserialized.
Auto Trait Implementations§
impl<B, const D: usize> Freeze for RmsPropState<B, D>where
SquareAvgState<B, D>: Freeze,
CenteredState<B, D>: Freeze,
Option<RmsPropMomentumState<B, D>>: Freeze,
impl<B, const D: usize> RefUnwindSafe for RmsPropState<B, D>where
SquareAvgState<B, D>: RefUnwindSafe,
CenteredState<B, D>: RefUnwindSafe,
Option<RmsPropMomentumState<B, D>>: RefUnwindSafe,
impl<B, const D: usize> Send for RmsPropState<B, D>where
SquareAvgState<B, D>: Send,
CenteredState<B, D>: Send,
Option<RmsPropMomentumState<B, D>>: Send,
impl<B, const D: usize> Sync for RmsPropState<B, D>where
SquareAvgState<B, D>: Sync,
CenteredState<B, D>: Sync,
Option<RmsPropMomentumState<B, D>>: Sync,
impl<B, const D: usize> Unpin for RmsPropState<B, D>where
SquareAvgState<B, D>: Unpin,
CenteredState<B, D>: Unpin,
Option<RmsPropMomentumState<B, D>>: Unpin,
impl<B, const D: usize> UnsafeUnpin for RmsPropState<B, D>where
SquareAvgState<B, D>: UnsafeUnpin,
CenteredState<B, D>: UnsafeUnpin,
Option<RmsPropMomentumState<B, D>>: UnsafeUnpin,
impl<B, const D: usize> UnwindSafe for RmsPropState<B, D>where
SquareAvgState<B, D>: UnwindSafe,
CenteredState<B, D>: UnwindSafe,
Option<RmsPropMomentumState<B, D>>: UnwindSafe,
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more