diff --git a/pytorch_lightning/metrics/functional/classification.py b/pytorch_lightning/metrics/functional/classification.py index d91e5ebae1040..a0466d6c5ba93 100644 --- a/pytorch_lightning/metrics/functional/classification.py +++ b/pytorch_lightning/metrics/functional/classification.py @@ -387,4 +387,4 @@ def dice_score(pred: torch.Tensor, target: torch.Tensor, bg: bool = False, score_cls = (2 * tp).to(torch.float) / denom scores[i] += score_cls - return reduce(scores, reduction=reduction) \ No newline at end of file + return reduce(scores, reduction=reduction)