LSQ

class LSQ(dtype, enable=True, eps=1e-05, **kwargs)

LSQ：https://arxiv.org/pdf/1902.08153.pdf 在每个权重和激活层的量化器步长处评估和缩放任务的梯度损失

参数

• dtype (Union[str, QuantDtypeMeta]) – 指明量化输入数据类型的字符串或 QuantDtypeMeta.

• enable (bool) – 选择 normal_forward 或是 fake_quant_forward.

• eps (float) – 为了避免除零而设置的一个很小的值。默认: 1e-5

上一页

TQT

下一页

megengine.quantization.quantize_qat