search

kevinzakka / pytorch-goodies

PyTorch Boilerplate For Research
603 stars 71 forks source link

Typo in dice_loss method #2

Closed AnshumaanDash closed 5 years ago

AnshumaanDash commented 5 years ago

Hi, Thanks for the awesome repository.

I happened to notice the following typo in the dice_loss method in losses.py

line 84: probas = F.softmax(probas, dim=1)

It should be probas = F.softmax(logits, dim=1)

Kindly make the change. Thanks

kevinzakka commented 5 years ago

@AnshumaanDash thanks for the heads-up, I had that fixed locally (along with a GPU fix) but never pushed them. Should be good to go now!