An implementation of focal loss in pytorch meant to be understandable and easily swappable with nn.functional.cross_entropy and nn.CrossEntropyLoss
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool