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