diff --git a/mmdet/models/losses/focal_loss.py b/mmdet/models/losses/focal_loss.py index 57b28b04339d8ce5807f37622f106df6456a6e48..12b17df082d81acae2ccf9cd6c8211b762551069 100644 --- a/mmdet/models/losses/focal_loss.py +++ b/mmdet/models/losses/focal_loss.py @@ -13,7 +13,7 @@ class FocalLoss(nn.Module): gamma=2.0, alpha=0.25): super(FocalLoss, self).__init__() - assert use_sigmoid is True, 'Only sigmoid focaloss supported now.' + assert use_sigmoid is True, 'Only sigmoid focal loss supported now.' self.use_sigmoid = use_sigmoid self.loss_weight = loss_weight self.gamma = gamma