提交 cde9727a 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: 4aa38369264d69f4809db22f71962ae3c481c489
上级 94796060
......@@ -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
......
......@@ -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"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册