Implementation of focal loss in pytorch for unbalanced classification. - View it on GitHub