Browse Source

fix(mge/quantization): fix `qat.conv_bn` parameter issue

GitOrigin-RevId: 4aa3836926
release-1.2
Megvii Engine Team 4 years ago
parent
commit
cde9727a4f
2 changed files with 2 additions and 3 deletions
  1. +2
    -2
      imperative/python/megengine/module/qat/conv_bn.py
  2. +0
    -1
      imperative/python/megengine/module/quantized/linear.py

+ 2
- 2
imperative/python/megengine/module/qat/conv_bn.py View File

@@ -153,8 +153,8 @@ class _ConvBnActivation2d(Float._ConvBnActivation2d, QATModule):
float_module.conv.dilation, float_module.conv.dilation,
float_module.conv.groups, float_module.conv.groups,
float_module.conv.bias is not None, 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.weight = float_module.conv.weight
qat_module.conv.bias = float_module.conv.bias qat_module.conv.bias = float_module.conv.bias


+ 0
- 1
imperative/python/megengine/module/quantized/linear.py View File

@@ -37,7 +37,6 @@ class Linear(QuantizedModule):
ret = ret if self.output_dtype is None else ret.astype(self.output_dtype) ret = ret if self.output_dtype is None else ret.astype(self.output_dtype)
return ret return ret



@classmethod @classmethod
def from_qat_module(cls, qat_module: QAT.Linear): def from_qat_module(cls, qat_module: QAT.Linear):
r""" r"""


Loading…
Cancel
Save