Struct Fp32MasterState 

pub struct Fp32MasterState<B: Backend, const D: usize, S: Record<B> + Clone> {
    pub master: Tensor<B, D>,
    pub inner: Option<S>,
}

Master parameter and the original optimizer’s state, recorded together. Use full-precision recorder settings to preserve the master for continuation.

Fields

master: Tensor<B, D>

Authoritative FP32 parameter after the most recent update.

inner: Option<S>

State returned by the wrapped optimizer.

Trait Implementations

impl<B: Clone + Backend, const D: usize, S: Clone + Record<B> + Clone> Clone for Fp32MasterState<B, D, S>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B: Backend, const D: usize, S: Record<B> + Clone> Record<B> for Fp32MasterState<B, D, S>

type Item<P: PrecisionSettings> = <(Tensor<B, D>, Option<S>) as Record<B>>::Item<P>

Type of the item that can be serialized and deserialized.

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

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

fn from_item<P: PrecisionSettings>( item: Self::Item<P>, device: &B::Device, ) -> Self

Convert the given item into a record.