Browse Source

fix symmetric bug

tags/v1.1.0
xiaoyisd 5 years ago
parent
commit
5ea8e7fef7
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

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


Loading…
Cancel
Save