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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

daveboat/pytorch_focal_loss Stargazers