CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
MIT License
2.17k stars 374 forks source link

new feature request: dice loss #17

Closed feng-1985 closed 3 years ago

feng-1985 commented 3 years ago

Dice Loss for Data-imbalanced NLP Tasks https://arxiv.org/pdf/1911.02855.pdf https://github.com/ShannonAI/mrc-for-flat-nested-ner/blob/master/loss/dice_loss.py