Enum PrecisionStrategy 

pub enum PrecisionStrategy {
    FP32,
    FP16,
    BF16,
    Mixed {
        forward_precision: String,
        backward_precision: String,
        loss_scaling: bool,
    },
    Quantized {
        weight_bits: u8,
        activation_bits: u8,
        quantization_method: String,
    },
}

Precision strategies for different hardware

Variants

FP32

Full precision (FP32)

FP16

Half precision (FP16)

BF16

Brain floating point (BF16)

Mixed

Mixed precision training

Fields

forward_precision: String

Forward pass precision

backward_precision: String

Backward pass precision

loss_scaling: bool

Enable loss scaling

Quantized

Integer quantization

Fields

weight_bits: u8

Number of bits for weights

activation_bits: u8

Number of bits for activations

quantization_method: String

Quantization method

Trait Implementations

impl Clone for PrecisionStrategy

fn clone(&self) -> PrecisionStrategy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PrecisionStrategy

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

Formats the value using the given formatter.