diff --git a/python_module/megengine/module/qat/conv_bn.py b/python_module/megengine/module/qat/conv_bn.py index b62270b6..38e12052 100644 --- a/python_module/megengine/module/qat/conv_bn.py +++ b/python_module/megengine/module/qat/conv_bn.py @@ -62,6 +62,7 @@ class _ConvBnActivation2d(Float._ConvBnActivation2d, QATModule): self.conv.groups, -1, 1, 1, 1 ) + w_fold = self.apply_quant_weight(w_fold) # b_fold = gamma * (b - bn_mean) / bn_std + beta b_fold = beta + gamma * (conv_bias - bn_mean) * bn_istd return w_fold, b_fold