mindspore_gs.quantization

量化感知训练算法集。

import mindspore_gs.quantization as qat

Simulated Quantization算法

mindspore_gs.quantization.SimulatedQuantizationAwareTraining

模拟量化感知训练的基本实现,该算法在训练时使用伪量化节点来模拟量化计算的损失,并通过反向传播更新网络参数,使得网络参数更好地适应量化带来的损失。

SLB量化算法

mindspore_gs.quantization.SlbQuantAwareTraining

SLB(Searching for Low-Bit Weights in Quantized Neural Networks)算法的实现,该算法将量化神经网络中的离散权重作为可搜索的变量,并实现了一种微分方法去精确的实现该搜索。