Fixed math error in DiceLoss backward pass (#35)

Former-commit-id: 649124b3bd26272bb8442843a546c4e895ec14ff
2018-10-26 18:53:05 +02:00 · 2018-10-26 18:53:05 +02:00 · 7e5a8727de
parent f2d88f98a7
commit 7e5a8727de
1 changed files with 1 additions and 1 deletions
--- a/dice_loss.py
+++ b/dice_loss.py
@ -20,7 +20,7 @@ class DiceCoeff(Function):
        grad_input = grad_target = None

        if self.needs_input_grad[0]:
-            grad_input = grad_output * 2 * (target * self.union + self.inter) \
+            grad_input = grad_output * 2 * (target * self.union - self.inter) \
                         / self.union * self.union
        if self.needs_input_grad[1]:
            grad_target = None