Fix bug NLLLoss

predict-idlab · Oct 3, 2024 · 69b1ae9 · 69b1ae9
1 parent 5a992f6
commit 69b1ae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/landmarker/losses/losses.py b/src/landmarker/losses/losses.py
@@ -575,7 +575,7 @@ def forward(self, output, target):
             output = self.log_softmax(output)
         else:
             output = torch.log(output.double())
-        nll = -target * torch.log(output.double())
+        nll = -target * output.double()
         if self.spatial_dims == 2:
             dim = (2, 3)
         else: