Struct MixedPrecisionConfig 

pub struct MixedPrecisionConfig {
    pub enabled: bool,
    pub loss_scale: Float,
    pub dynamic_loss_scaling: bool,
    pub initial_loss_scale: Float,
    pub growth_factor: Float,
    pub backoff_factor: Float,
    pub growth_interval: usize,
    pub fp32_operations: Vec<String>,
    pub use_amp: bool,
    pub gradient_clip_threshold: Option<Float>,
}

Mixed precision configuration

Fields

enabled: bool

Enable mixed precision training

loss_scale: Float

Loss scaling factor to prevent gradient underflow

dynamic_loss_scaling: bool

Dynamic loss scaling

initial_loss_scale: Float

Initial loss scale for dynamic scaling

growth_factor: Float

Growth factor for loss scaling

backoff_factor: Float

Backoff factor when overflow is detected

growth_interval: usize

Number of steps without overflow before increasing scale

fp32_operations: Vec<String>

Operations to keep in FP32 (for numerical stability)

use_amp: bool

Use automatic mixed precision (AMP)

gradient_clip_threshold: Option<Float>

Gradient clipping threshold

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.