From 5ea8e7fef7a4f2f45a8040751f4898772f7b0eb3 Mon Sep 17 00:00:00 2001 From: xiaoyisd Date: Thu, 3 Dec 2020 20:12:28 +0800 Subject: [PATCH] fix symmetric bug --- model_zoo/official/cv/mobilenetv2_quant/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/mobilenetv2_quant/train.py b/model_zoo/official/cv/mobilenetv2_quant/train.py index 9d883cf4313..ede5f5d703a 100644 --- a/model_zoo/official/cv/mobilenetv2_quant/train.py +++ b/model_zoo/official/cv/mobilenetv2_quant/train.py @@ -164,7 +164,7 @@ def train_on_gpu(): # convert fusion network to quantization aware network quantizer = QuantizationAwareTraining(bn_fold=True, per_channel=[True, False], - symmetric=[True, False], + symmetric=[False, False], freeze_bn=1000000, quant_delay=step_size * 2) network = quantizer.quantize(network)