Struct QatConfig 

pub struct QatConfig {
    pub target: QuantTarget,
    pub scheme: QuantScheme,
    pub rounding: RoundingMode,
    pub lr: f64,
    pub beta1: f64,
    pub beta2: f64,
    pub eps: f64,
    pub weight_decay: f64,
}

Configuration for a QatOptimizer: the quantization target plus the (Adam/AdamW) master-update hyper-parameters.

Fields

target: QuantTarget

Quantization target for the fake-quant view.

scheme: QuantScheme

Integer scheme (ignored for fp8 targets).

rounding: RoundingMode

Rounding mode applied when deriving the quantized view.

lr: f64

Learning rate.

beta1: f64

First-moment decay (beta1).

beta2: f64

Second-moment decay (beta2).

eps: f64

Numerical-stability epsilon.

weight_decay: f64

Decoupled (AdamW) weight decay.

Implementations

impl QatConfig

pub fn new( target: QuantTarget, scheme: QuantScheme, rounding: RoundingMode, lr: f64, ) -> Self

Construct a config with standard Adam defaults (beta1 = 0.9, beta2 = 0.999, eps = 1e-8, weight_decay = 0).

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 Copy for QatConfig

impl Debug for QatConfig

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

Formats the value using the given formatter.

impl PartialEq for QatConfig

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for QatConfig