diff --git a/torchmetrics/detection/map.py b/torchmetrics/detection/map.py index 0bd7bc4e37c..ccd13c61d28 100644 --- a/torchmetrics/detection/map.py +++ b/torchmetrics/detection/map.py @@ -786,4 +786,5 @@ def compute(self) -> dict: metrics.update(mar) metrics.map_per_class = map_per_class_values metrics[f"mar_{self.max_detection_thresholds[-1]}_per_class"] = mar_max_dets_per_class_values + return metrics