-
Notifications
You must be signed in to change notification settings - Fork 27.4k
Dice Loss PR #1249
Copy link
Copy link
Open
Labels
StaleenhancementNot as big of a feature, but technically not a bug. Should be easy to fixNot as big of a feature, but technically not a bug. Should be easy to fixmodule: lossProblem is related to loss functionProblem is related to loss functiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
Hi, I have implemented a Dice loss function which is used in segmentation tasks, and sometimes even preferred over cross_entropy. More info in this paper:
http://campar.in.tum.de/pub/milletari2016Vnet/milletari2016Vnet.pdf
Here's the link of it:
https://github.com/rogertrullo/pytorch/blob/rogertrullo-dice_loss/torch/nn/functional.py#L708
How could I submit a PR?
thanks!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
StaleenhancementNot as big of a feature, but technically not a bug. Should be easy to fixNot as big of a feature, but technically not a bug. Should be easy to fixmodule: lossProblem is related to loss functionProblem is related to loss functiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module