Browse Source

use one_conv_fold for mobilenetv2_quant

tags/v1.1.0
yuchaojie 5 years ago
parent
commit
a47431a25a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/mobilenetv2_quant/train.py

+ 1
- 1
model_zoo/official/cv/mobilenetv2_quant/train.py View File

@@ -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


Loading…
Cancel
Save