fix symmetric bug

This commit is contained in:
xiaoyisd 2020-12-03 20:12:28 +08:00
parent 6b9e402790
commit 5ea8e7fef7
1 changed files with 1 additions and 1 deletions

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)