jaredfern / pytorch-multi-class-focal-loss Goto Github PK
View Code? Open in Web Editor NEWThis project forked from adeelh/pytorch-multi-class-focal-loss
An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
License: MIT License