removed redundant line in dice loss (#710)

kaiko-ai · Dec 6, 2024 · 9f73e79 · 9f73e79
1 parent 4b09f96
commit 9f73e79
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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)