Struct PrecisionConfig 

pub struct PrecisionConfig {
    pub default_precision: Precision,
    pub auto_mixed_precision: bool,
    pub loss_scale: Option<f64>,
    pub dynamic_loss_scaling: bool,
}

Precision configuration for an executor.

Fields

default_precision: Precision

Default precision for tensors

auto_mixed_precision: bool

Enable automatic mixed precision

loss_scale: Option<f64>

Loss scaling for mixed precision training

dynamic_loss_scaling: bool

Dynamic loss scaling (adjust based on gradients)

Implementations

impl PrecisionConfig

pub fn f32() -> Self

Create a configuration for FP32 precision.

pub fn f64() -> Self

Create a configuration for FP64 precision.

pub fn mixed_precision() -> Self

Create a configuration for mixed precision training.

pub fn with_auto_mixed_precision(self, enable: bool) -> Self

Enable automatic mixed precision.

pub fn with_loss_scale(self, scale: f64) -> Self

Set the loss scale for mixed precision training.

pub fn with_dynamic_loss_scaling(self, enable: bool) -> Self

Enable dynamic loss scaling.

Trait Implementations

impl Clone for PrecisionConfig

fn clone(&self) -> PrecisionConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PrecisionConfig

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

Formats the value using the given formatter.

impl Default for PrecisionConfig

fn default() -> Self

Returns the “default value” for a type.