diff --git a/python_module/megengine/module/qat/conv_bn.py b/python_module/megengine/module/qat/conv_bn.py index b62270b61d76670adfc234355634cb4fdb2f2ec7..38e12052171b2be9b096ca1ab64fbf504b672d15 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