Fix bug from last PR in histogram collector (#1370)

Co-authored-by: liord <[email protected]>
sony · Mar 2, 2025 · de1450a · de1450a
1 parent 850aa63
commit de1450a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_compression_toolkit/core/common/collectors/histogram_collector.py b/model_compression_toolkit/core/common/collectors/histogram_collector.py
@@ -78,7 +78,7 @@ def _merge_histograms(self):
             merged_histogram_counts = None
             for histogram in self._histogram_per_iteration:  # Iterate all collected histograms and merge them
                 if merged_histogram_counts is None:  # First histogram to consider
-                    merged_histogram_counts = histogram[0].astype(np.float64)  # Convert to float64
+                    merged_histogram_counts = interpolate_histogram(merged_histogram_bins, histogram[1], histogram[0])
                 else:  # Merge rest of histograms into existing final histogram
                     merged_histogram_counts += interpolate_histogram(merged_histogram_bins, histogram[1], histogram[0])