FocalLoss() gamma and alpha default values

2020-03-13 16:51:30 -07:00
parent 208b9a73fe
commit 418269d739
2 changed files with 2 additions and 2 deletions
@@ -340,7 +340,7 @@ def wh_iou(wh1, wh2):
 class FocalLoss(nn.Module):
    # Wraps focal loss around existing loss_fcn() https://arxiv.org/pdf/1708.02002.pdf
    # i.e. criteria = FocalLoss(nn.BCEWithLogitsLoss(), gamma=2.5)
-    def __init__(self, loss_fcn, gamma=0.5, alpha=1):
+    def __init__(self, loss_fcn, gamma=1.5, alpha=0.25):
        super(FocalLoss, self).__init__()
        self.loss_fcn = loss_fcn
        self.gamma = gamma