From a47431a25a4d388218bde29035b6e8aec065e5fc Mon Sep 17 00:00:00 2001 From: yuchaojie Date: Fri, 11 Dec 2020 10:43:30 +0800 Subject: [PATCH] use one_conv_fold for mobilenetv2_quant --- 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 5fcaa9ba2f9..65a03bae318 100644 --- a/model_zoo/official/cv/mobilenetv2_quant/train.py +++ b/model_zoo/official/cv/mobilenetv2_quant/train.py @@ -102,7 +102,7 @@ def train_on_ascend(): quantizer = QuantizationAwareTraining(bn_fold=True, per_channel=[True, False], symmetric=[True, False], - one_conv_fold=False) + one_conv_fold=True) network = quantizer.quantize(network) # get learning rate