daveboat/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
No issues in this repository yet.
An implementation of focal loss in pytorch meant to be understandable and easily swappable with nn.functional.cross_entropy and nn.CrossEntropyLoss
Python
No issues in this repository yet.