Add SoftMax Focal loss - Githubissues

bonlime / pytorch-tools

Tool box for PyTorch

MIT License

186 stars 16 forks source link

Closed bonlime closed 2 years ago

bonlime commented 4 years ago

Текущая реализация поддерживает только binary_cross_entropy для focal лосс, что соответствует сигмоиде. хочется вариант Focal для softmax