Struct QATConfig 

pub struct QATConfig {
    pub weight_scheme: QuantizationScheme,
    pub activation_scheme: ActivationQuantScheme,
    pub symmetric: bool,
    pub warmup_epochs: usize,
    pub schedule: QATSchedule,
    pub quantize_first_last: bool,
    pub observer_config: ObserverConfig,
    pub use_ste: bool,
}

QAT training configuration

Fields

weight_scheme: QuantizationScheme

Weight quantization scheme

activation_scheme: ActivationQuantScheme

Activation quantization scheme

symmetric: bool

Whether to use symmetric quantization

warmup_epochs: usize

Number of warmup epochs before enabling quantization

schedule: QATSchedule

QAT schedule for gradual quantization introduction

quantize_first_last: bool

Whether to quantize first and last layers

observer_config: ObserverConfig

Observer configuration

use_ste: bool

Whether to use straight-through estimator

Trait Implementations

impl Clone for QATConfig

fn clone(&self) -> QATConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for QATConfig

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

Formats the value using the given formatter.

impl Default for QATConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for QATConfig

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for QATConfig

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

where __S: Serializer,

Serialize this value into the given Serde serializer.