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!