PolyLoss implementation using PyTorch Install pip install polyloss Usage from polyloss import PolyLoss ... criterion = PolyLoss(epsilon=2.0) ...