saarques/pytorch_focal_loss
An implementation of focal loss in pytorch meant to be understandable and easily swappable with nn.functional.cross_entropy and nn.CrossEntropyLoss
Python
An implementation of focal loss in pytorch meant to be understandable and easily swappable with nn.functional.cross_entropy and nn.CrossEntropyLoss
Python