From 7d4faf7fc175d0acd6a31a66dc08f068b0c023b3 Mon Sep 17 00:00:00 2001 From: Tian Pang Date: Fri, 14 Mar 2025 13:54:24 +0800 Subject: [PATCH] fix normalize function in entropy calibrator Signed-off-by: Tian Pang --- .../pytorch_quantization/calib/histogram.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/pytorch-quantization/pytorch_quantization/calib/histogram.py b/tools/pytorch-quantization/pytorch_quantization/calib/histogram.py index 376fff569..0c52df4f7 100644 --- a/tools/pytorch-quantization/pytorch_quantization/calib/histogram.py +++ b/tools/pytorch-quantization/pytorch_quantization/calib/histogram.py @@ -189,7 +189,7 @@ def _compute_amax_entropy(calib_hist, calib_bin_edges, num_bits, unsigned, strid def _normalize_distr(distr): summ = np.sum(distr) if summ != 0: - distr = distr / summ + distr /= summ bins = calib_hist[:] bins[0] = bins[1]