!9817 use one_conv_fold for mobilenetv2_quant

From: @yuchaojie Reviewed-by: @linqingke,@xu-yfei Signed-off-by: @linqingke
2020-12-11 15:20:30 +08:00 · 2020-12-11 15:20:30 +08:00 · 1cf093ed0c
parent b5ab80d125 a47431a25a
commit 1cf093ed0c
1 changed files with 1 additions and 1 deletions
--- 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