Remove unnecessary IoU ignore_index comment

Lightning-AI · Sep 15, 2020 · 8b84b04 · 8b84b04
1 parent f0cc0fa
commit 8b84b04
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pytorch_lightning/metrics/functional/classification.py b/pytorch_lightning/metrics/functional/classification.py
@@ -1010,7 +1010,6 @@ def iou(
     scores = torch.zeros(num_classes, device=pred.device, dtype=torch.float32)
 
     for class_idx in range(num_classes):
-        # Skip this class if its index is being ignored.
         if class_idx == ignore_index:
             continue