We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 850aa63 commit de1450aCopy full SHA for de1450a
model_compression_toolkit/core/common/collectors/histogram_collector.py
@@ -78,7 +78,7 @@ def _merge_histograms(self):
78
merged_histogram_counts = None
79
for histogram in self._histogram_per_iteration: # Iterate all collected histograms and merge them
80
if merged_histogram_counts is None: # First histogram to consider
81
- merged_histogram_counts = histogram[0].astype(np.float64) # Convert to float64
+ merged_histogram_counts = interpolate_histogram(merged_histogram_bins, histogram[1], histogram[0])
82
else: # Merge rest of histograms into existing final histogram
83
merged_histogram_counts += interpolate_histogram(merged_histogram_bins, histogram[1], histogram[0])
84
0 commit comments