removed redundant line in dice loss

kaiko-ai · Nov 20, 2024 · ec12e40 · ec12e40
1 parent 50c90a3
commit ec12e40
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)