mindspore_gs.quantization

Quantization aware training algorithms.

import mindspore_gs.quantization as qat

Simulated Quantization Algorithm

mindspore_gs.quantization.SimulatedQuantizationAwareTraining

Basic implementation of simulated quantization aware training, this algorithm adopts fake quantizer to simulate the loss of quantization calculation, and network parameters are updated through backpropagation, so that the network parameters can better adapt to the loss caused by quantization.

SLB Algorithm

mindspore_gs.quantization.SlbQuantAwareTraining

Implementation of slb quantization algorithm, this algorithm regards the discrete weights in an arbitrary quantized neural network as searchable variables, and utilize a differential method to search them accurately.