forked from mindspore-Ecosystem/mindspore
!9453 fix symmetric bug
From: @xiaoyisd Reviewed-by: @liangchenghui,@jjfeing Signed-off-by: @liangchenghui
This commit is contained in:
commit
9aae337f11
|
@ -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