Fix a typo

pytorch · Sep 4, 2024 · 8bfb802 · 8bfb802
1 parent 248fe89
commit 8bfb802
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/metrics/vision/object_detection_average_precision_recall.py b/ignite/metrics/vision/object_detection_average_precision_recall.py
@@ -104,7 +104,7 @@ def box_iou(pred_boxes: torch.Tensor, gt_boxes: torch.Tensor, iscrowd: torch.Boo
         except ImportError:
             raise ModuleNotFoundError("This metric requires torchvision to be installed.")
 
-        precision = torch.double if not torch.device(device) != torch.device("mps") else torch.float32
+        precision = torch.double if torch.device(device) != torch.device("mps") else torch.float32
 
         if iou_thresholds is None:
             iou_thresholds = torch.linspace(0.5, 0.95, 10, device=device, dtype=precision)