This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.x by this push:
new f995e51 Fix to quantization dshape bug (#19501)
f995e51 is described below
commit f995e51b73b5ba2076acaca4e1e993afcd5abf1a
Author: Adam <[email protected]>
AuthorDate: Tue Nov 10 20:17:20 2020 +0100
Fix to quantization dshape bug (#19501)
---
python/mxnet/contrib/quantization.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/mxnet/contrib/quantization.py
b/python/mxnet/contrib/quantization.py
index 0305adb..1422e07 100644
--- a/python/mxnet/contrib/quantization.py
+++ b/python/mxnet/contrib/quantization.py
@@ -893,8 +893,8 @@ def quantize_net_v2(network, quantized_dtype='auto',
quantize_mode='full', quant
dshapes = calib_data.provide_data
else:
calib_data, dshapes = _as_data_iter(calib_data)
- if not data_shapes:
- data_shapes = dshapes
+ if not data_shapes:
+ data_shapes = dshapes
if not data_shapes:
raise ValueError('data_shapes required')
data_nd = []
