Struct RmsPropState
pub struct RmsPropState<B, const D: usize>where
B: Backend,{
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§
§
impl<B, const D: usize>
RmsPropState<B, D>where
B: Backend,
impl<B, const D: usize>
RmsPropState<B, D>where
B: Backend,
pub fn new(
square_avg: SquareAvgState<B,
D>,
centered: CenteredState<B, D>,
momentum: Option<RmsPropMomentumState<B,
D>>,
) -> RmsPropState<B, D>
pub fn new( square_avg: SquareAvgState<B, D>, centered: CenteredState<B, D>, momentum: Option<RmsPropMomentumState<B, D>>, ) -> RmsPropState<B, D>
Constructs a new RmsPropState
.
Trait Implementations§
§
impl<B, const D: usize>
Clone for RmsPropState<B, D>
impl<B, const D: usize> Clone for RmsPropState<B, D>
§
fn clone(&self) -> RmsPropState<B, D>
fn clone(&self) -> RmsPropState<B, D>
Returns a copy of the value. Read
more
1.0.0 · Source§
fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source
. Read
more
§
impl<B, const D: usize>
Record<B> for RmsPropState<B, D>where
B: Backend,
impl<B, const D: usize>
Record<B> for RmsPropState<B, D>where
B: Backend,
§
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.
§
fn into_item<S>(self) -> <RmsPropState<B, D> as
Record<B>>::Item<S>where
S: PrecisionSettings,
fn into_item<S>(self) -> <RmsPropState<B, D> as
Record<B>>::Item<S>where
S: PrecisionSettings,
Convert the current record into the corresponding item that
follows the given settings.
§
fn from_item<S>(
item: <RmsPropState<B, D> as
Record<B>>::Item<S>,
device: &<B as Backend>::Device,
) -> RmsPropState<B, D>
where
S: PrecisionSettings,
fn from_item<S>(
item: <RmsPropState<B, D> as
Record<B>>::Item<S>,
device: &<B as Backend>::Device,
) -> RmsPropState<B, D>
where
S: PrecisionSettings,
Convert the given item into a record.
Auto Trait Implementations§
impl<B, const D: usize> Freeze for RmsPropState<B, D>
where
<B as Backend>::FloatTensorPrimitive:
Freeze,
<B as Backend>::QuantizedTensorPrimitive:
Freeze,
impl<B, const D: usize> RefUnwindSafe for RmsPropState<B, D>
where
<B as Backend>::FloatTensorPrimitive:
RefUnwindSafe,
<B as Backend>::QuantizedTensorPrimitive:
RefUnwindSafe,
impl<B, const D: usize> Send for RmsPropState<B, D>
impl<B, const D: usize> Sync for RmsPropState<B, D>
impl<B, const D: usize> Unpin for RmsPropState<B, D>
impl<B, const D: usize> UnwindSafe for RmsPropState<B, D>
where
<B as Backend>::FloatTensorPrimitive:
UnwindSafe,
<B as Backend>::QuantizedTensorPrimitive:
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
Source§
impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§
impl<T> Instrument for T
impl<T> Instrument for T
§
fn instrument(self, span: Span) ->
Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§
fn in_current_span(self) ->
Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§
impl<T> IntoEither for T
impl<T> IntoEither for T
Source§
fn into_either(self, into_left: bool)
-> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left
is true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read
more
Source§
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left(&self)
returns true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read
more