From 0d0968557f864b316a9817988d55d9998233ae40 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 8 Mar 2021 14:25:58 +0800 Subject: [PATCH] fix(imperative/quantization): add conditional when passing parameters name GitOrigin-RevId: f38042aeabd653cbc23a744de581b550135c0a86 --- imperative/python/megengine/module/quantized/conv_bn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/module/quantized/conv_bn.py b/imperative/python/megengine/module/quantized/conv_bn.py index 4b3b5772..2bfc070a 100644 --- a/imperative/python/megengine/module/quantized/conv_bn.py +++ b/imperative/python/megengine/module/quantized/conv_bn.py @@ -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 -- GitLab