Fix a bug

rahul003 · Jan 16, 2018 · f7853f2 · f7853f2
1 parent 70b9e38
commit f7853f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/quantization.py b/python/mxnet/quantization.py
@@ -273,7 +273,7 @@ def _get_optimal_thresholds(nd_dict, num_bins=8001, num_quantized_bins=255, logg
         logger.info('Calculating optimal thresholds for quantization using KL divergence'
                     ' with num_bins=%d and num_quantized_bins=%d' % (num_bins, num_quantized_bins))
     th_dict = {}
-    for k, v in nd_dict:
+    for k, v in nd_dict.items():
         min_val, max_val, min_divergence, opt_th = _get_optimal_threshold(v, num_bins=num_bins,
                                                                           num_quantized_bins=num_quantized_bins)
         del v  # release the memory of ndarray