/pytorch-multi-class-focal-loss

A simple pytorch implementation for multi-class focal loss

Primary LanguagePython

pytorch-multi-class-focal-loss

A simple pytorch implementation for multi-class focal loss

Someone told me that this implementation seems to be wrong, so it's better not to use it directly.