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

Another error in diceloss

Browse source 
Former-commit-id: 24a26eaa42b2a8e24335a421047af37cd11adb85
This commit is contained in:
milesial 2018-10-28 14:54:14 +01:00 committed by GitHub
parent 7e5a8727de
commit 46d1db3115
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

@ -21,7 +21,7 @@ class DiceCoeff(Function):
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