Enum PrecisionMode 

pub enum PrecisionMode {
    Full,
    Half,
    BFloat16,
}

Precision mode for training.

Variants

Full

Full precision (FP32)

Half

Half precision (FP16)

BFloat16

Brain float 16 (BF16)

Implementations

impl PrecisionMode

pub fn to_dtype(&self) -> DType

Convert precision mode to Candle DType.

pub fn from_dtype(dtype: DType) -> Result<PrecisionMode, UnslothError>

Get the precision mode from a Candle DType.

Errors

Returns error if dtype is not a supported floating point type.

Trait Implementations

impl Clone for PrecisionMode

fn clone(&self) -> PrecisionMode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PrecisionMode

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

Formats the value using the given formatter.

impl PartialEq for PrecisionMode

fn eq(&self, other: &PrecisionMode) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for PrecisionMode

impl Eq for PrecisionMode

impl StructuralPartialEq for PrecisionMode