Remove unnecessary IoU ignore_index comment

Lightning-AI · Sep 14, 2020 · a63d403 · a63d403
1 parent 2579475
commit a63d403
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
@@ -1016,7 +1016,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