From 9f73e7932fcddae4d7b9d08292f17fd056377804 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicolas=20K=C3=A4nzig?= <36882833+nkaenzig@users.noreply.github.com> Date: Fri, 6 Dec 2024 10:16:37 +0100 Subject: [PATCH] removed redundant line in dice loss (#710) --- src/eva/vision/losses/dice.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/eva/vision/losses/dice.py b/src/eva/vision/losses/dice.py index 8e6133b3..d5d31d17 100644 --- a/src/eva/vision/losses/dice.py +++ b/src/eva/vision/losses/dice.py @@ -45,9 +45,6 @@ def forward(self, inputs: torch.Tensor, targets: torch.Tensor) -> torch.Tensor: inputs, targets = _apply_ignore_index(inputs, targets, self.ignore_index) targets = _to_one_hot(targets, num_classes=inputs.shape[1]) - if targets.ndim == 3: - targets = one_hot(targets[:, None, ...], num_classes=inputs.shape[1]) - return super().forward(inputs, targets)