Enum QATSchedule 

pub enum QATSchedule {
    Immediate,
    Gradual {
        start_epoch: usize,
        end_epoch: usize,
        weight_schedule: GradualSchedule,
        activation_schedule: GradualSchedule,
    },
    LayerWise {
        schedule: HashMap<String, LayerSchedule>,
    },
    Progressive {
        start_bits: u8,
        end_bits: u8,
        reduction_epochs: Vec<usize>,
    },
}

QAT training schedule

Variants

Immediate

Immediate quantization from start

Gradual

Gradual introduction over epochs

Fields

start_epoch: usize

end_epoch: usize

weight_schedule: GradualSchedule

activation_schedule: GradualSchedule

LayerWise

Custom layer-by-layer schedule

Fields

schedule: HashMap<String, LayerSchedule>

Progressive

Progressive bit reduction

Fields

start_bits: u8

end_bits: u8

reduction_epochs: Vec<usize>

Trait Implementations

impl Clone for QATSchedule

fn clone(&self) -> QATSchedule

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for QATSchedule

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for QATSchedule

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for QATSchedule

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.