From 3e58cbb8d25a01d17c86b3fcff00735421ec1f01 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 6 Aug 2021 17:10:22 +0800 Subject: [PATCH] fix(mge/quantization): fix lsq init format error GitOrigin-RevId: 032de9c7738622fd6d83e706e5a5b85355645dfd --- imperative/python/megengine/quantization/fake_quant.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/quantization/fake_quant.py b/imperative/python/megengine/quantization/fake_quant.py index c1469d33f..f6625d708 100644 --- a/imperative/python/megengine/quantization/fake_quant.py +++ b/imperative/python/megengine/quantization/fake_quant.py @@ -132,7 +132,7 @@ class LSQ(_FakeQuantize, QParamsModuleMixin): :param eps:a small value to avoid division by zero. Default: 1e-5 """ - def init( + def __init__( self, dtype: Union[str, QuantDtypeMeta], enable: bool = True, @@ -142,6 +142,9 @@ class LSQ(_FakeQuantize, QParamsModuleMixin): super().__init__(dtype=dtype, enable=enable, **kwargs) self.eps = Tensor(eps, dtype="float32") self.step_size = Parameter(1.0, dtype="float32") + self.mode = None + self.zero_point = Tensor(0.0, dtype="float32") + self.grad_scale = Tensor(1.0, dtype="float32") def set_qparams(self, qparams: LSQParams): self.mode = qparams.mode -- GitLab