From 756236b6166d7d6e6b9b4b73a0efc2fcdb69573d Mon Sep 17 00:00:00 2001 From: lizz <innerlee@users.noreply.github.com> Date: Tue, 11 Jun 2019 14:43:48 +0800 Subject: [PATCH] typo (#782) Signed-off-by: innerlee <363664470@qq.com> --- mmdet/models/losses/focal_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/models/losses/focal_loss.py b/mmdet/models/losses/focal_loss.py index 57b28b0..12b17df 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 -- GitLab