Browse Source

fix(imperative/quantization): add conditional when passing parameters name

GitOrigin-RevId: f38042aeab
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
0d0968557f
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      imperative/python/megengine/module/quantized/conv_bn.py

+ 3
- 1
imperative/python/megengine/module/quantized/conv_bn.py View File

@@ -40,7 +40,9 @@ class _ConvBnActivation2d(Conv2d):
)
weight = w_fold.astype(qat_module.get_weight_dtype())
qconv.weight = Parameter(weight.numpy(), name=qat_module.conv.weight.name)
qconv.bias = Parameter(b_fold.numpy(), name=qat_module.conv.bias.name)
qconv.bias = Parameter(b_fold.numpy())
if qat_module.conv.bias is not None:
qconv.bias.name = qat_module.conv.bias.name
return qconv




Loading…
Cancel
Save