ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Fixed math error in DiceLoss backward pass (#35)

Browse source 
Former-commit-id: 649124b3bd26272bb8442843a546c4e895ec14ff
This commit is contained in:
milesial 2018-10-26 18:53:05 +02:00 committed by GitHub
parent f2d88f98a7
commit 7e5a8727de
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
dice_loss.py
View file

@ -20,7 +20,7 @@ class DiceCoeff(Function):
grad_input = grad_target = None grad_input = grad_target = None
if self.needs_input_grad[0]: 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 / self.union * self.union
if self.needs_input_grad[1]: if self.needs_input_grad[1]:
grad_target = None grad_target = None