Struct GradScalerCheckpoint 

pub struct GradScalerCheckpoint {
    pub scale: f32,
    pub growth_factor: f32,
    pub backoff_factor: f32,
    pub growth_interval: u32,
    pub consecutive_good_steps: u32,
    pub found_overflow: bool,
    pub enabled: bool,
}

Serialized GradScaler state.

Fields

scale: f32

Current loss scale.

growth_factor: f32

Factor applied after growth_interval good steps.

backoff_factor: f32

Factor applied after an overflow.

growth_interval: u32

Good steps between scale increases.

consecutive_good_steps: u32

Good steps since the last change.

found_overflow: bool

Whether the last step overflowed.

enabled: bool

Whether scaling is on.

Trait Implementations

impl Clone for GradScalerCheckpoint

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GradScalerCheckpoint

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GradScalerCheckpoint

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for GradScalerCheckpoint

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.