Skip to content

Commit

Permalink
Update compute description
Browse files Browse the repository at this point in the history
  • Loading branch information
vkinakh committed Aug 21, 2024
1 parent 143e4eb commit 8423ff0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torchmetrics/segmentation/mean_iou.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ def update(self, preds: Tensor, target: Tensor) -> None:
self.num_batches += 1

def compute(self) -> Tensor:
"""Update the state with the new data."""
"""Compute the final Mean Intersection over Union (mIoU)."""
return self.score / self.num_batches

def plot(self, val: Union[Tensor, Sequence[Tensor], None] = None, ax: Optional[_AX_TYPE] = None) -> _PLOT_OUT_TYPE:
Expand Down

0 comments on commit 8423ff0

Please sign in to comment.