LSQ#

class LSQ(dtype, enable=True, eps=1e-5, **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