Struct MixedPrecisionConfig 

pub struct MixedPrecisionConfig {
    pub compute_precision: Precision,
    pub weight_precision: Precision,
    pub initial_scale: f32,
    pub scale_growth_factor: f32,
    pub scale_backoff_factor: f32,
    pub scale_growth_interval: usize,
    pub dynamic_scaling: bool,
}

Configuration for mixed-precision training

Fields

compute_precision: Precision

Precision for activations and gradients

weight_precision: Precision

Precision for master weights (always fp32 recommended)

initial_scale: f32

Initial loss scale factor

scale_growth_factor: f32

Factor to increase scale by on successful step

scale_backoff_factor: f32

Factor to decrease scale by on overflow

scale_growth_interval: usize

Number of successful steps before increasing scale

dynamic_scaling: bool

Whether to use dynamic loss scaling

Implementations

impl MixedPrecisionConfig

pub fn fp32() -> Self

Create fp32 config (no mixed precision)

pub fn fp16() -> Self

Create fp16 mixed-precision config

pub fn bf16() -> Self

Create bf16 mixed-precision config

pub fn is_mixed(&self) -> bool

Check if mixed precision is enabled

pub fn with_initial_scale(self, scale: f32) -> Self

Set initial loss scale

pub fn with_dynamic_scaling(self, enabled: bool) -> Self

Enable/disable dynamic scaling

Trait Implementations

impl Clone for MixedPrecisionConfig

fn clone(&self) -> MixedPrecisionConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MixedPrecisionConfig

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

Formats the value using the given formatter.

impl Default for MixedPrecisionConfig

fn default() -> Self

Returns the “default value” for a type.