jahongir7174/PolyLoss

PolyLoss implementation using PyTorch

PolyLoss implementation using PyTorch

Install

pip install polyloss

Usage

from polyloss import PolyLoss

...
criterion = PolyLoss(epsilon=2.0)
...