An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
FocalLoss.forward()
always returns a tensor (#7)Full Changelog: https://github.com/AdeelH/pytorch-multi-class-focal-loss/compare/1.1...1.2
Minor changes.