Struct QuantizationAwareTraining

pub struct QuantizationAwareTraining { /* private fields */ }

Quantization-aware training (QAT) support

Implementations

impl QuantizationAwareTraining

pub fn new(config: QuantizationConfig) -> Self

Create new QAT instance

pub fn set_warmup_steps(&mut self, steps: usize)

Set warmup steps

pub fn init_layer_params( &mut self, layer_name: &str, tensor: &ArrayD<f32>, ) -> Result<()>

Initialize quantization parameters for a layer

pub fn fake_quantize( &mut self, layer_name: &str, tensor: &ArrayD<f32>, ) -> Result<ArrayD<f32>>

Apply fake quantization during training

pub fn get_quantization_params(&self) -> &HashMap<String, QuantizationParams>

Get final quantization parameters for deployment

pub fn add_quantization_noise( &self, tensor: &ArrayD<f32>, noise_scale: f32, ) -> ArrayD<f32>

Simulate quantization noise for better training

Trait Implementations

impl Debug for QuantizationAwareTraining

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

Formats the value using the given formatter.