From cde9727a4ffed966d7a0b5a564ac72740fb8aff5 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 17 Dec 2020 10:39:49 +0800 Subject: [PATCH] fix(mge/quantization): fix `qat.conv_bn` parameter issue GitOrigin-RevId: 4aa38369264d69f4809db22f71962ae3c481c489 --- imperative/python/megengine/module/qat/conv_bn.py | 4 ++-- imperative/python/megengine/module/quantized/linear.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/imperative/python/megengine/module/qat/conv_bn.py b/imperative/python/megengine/module/qat/conv_bn.py index 3de3509b..4ad64d04 100644 --- a/imperative/python/megengine/module/qat/conv_bn.py +++ b/imperative/python/megengine/module/qat/conv_bn.py @@ -153,8 +153,8 @@ class _ConvBnActivation2d(Float._ConvBnActivation2d, QATModule): float_module.conv.dilation, float_module.conv.groups, float_module.conv.bias is not None, - float_module.conv.conv_mode.name, - float_module.conv.compute_mode.name, + float_module.conv.conv_mode, + float_module.conv.compute_mode, ) qat_module.conv.weight = float_module.conv.weight qat_module.conv.bias = float_module.conv.bias diff --git a/imperative/python/megengine/module/quantized/linear.py b/imperative/python/megengine/module/quantized/linear.py index 1a817c5c..c3c3c330 100644 --- a/imperative/python/megengine/module/quantized/linear.py +++ b/imperative/python/megengine/module/quantized/linear.py @@ -37,7 +37,6 @@ class Linear(QuantizedModule): ret = ret if self.output_dtype is None else ret.astype(self.output_dtype) return ret - @classmethod def from_qat_module(cls, qat_module: QAT.Linear): r"""