Skip to content
Snippets Groups Projects
Commit 756236b6 authored by lizz's avatar lizz Committed by Kai Chen
Browse files

typo (#782)


Signed-off-by: default avatarinnerlee <363664470@qq.com>
parent ee82d58d
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment