forked from mindspore-Ecosystem/mindspore
fix symmetric bug
This commit is contained in:
parent
6b9e402790
commit
5ea8e7fef7
|
@ -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…
Reference in New Issue