Browse Source

zp=0

tags/v1.3.0
zhang__sss 4 years ago
parent
commit
15e2f2eb31
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      mindspore/compression/quant/quant_utils.py

+ 5
- 2
mindspore/compression/quant/quant_utils.py View File

@@ -74,8 +74,11 @@ def cal_quantization_params(input_min,
scale = (input_max - input_min) / (quant_max - quant_min)

# calculate zero point
zp_double = quant_min - input_min / scale
zp = np.floor(zp_double + 0.5)
if data_type == np.int8 and symmetric and not neg_trunc:
zp = np.zeros(input_min.shape)
else:
zp_double = quant_min - input_min / scale
zp = np.floor(zp_double + 0.5)

return scale, zp



Loading…
Cancel
Save