Skip to content
This repository was archived by the owner on Aug 14, 2025. It is now read-only.

Commit 21a996c

Browse files
committed
update comment
1 parent 0e17e04 commit 21a996c

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

FocalLoss/focal_loss.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ class BinaryFocalLoss(nn.Module):
99
This is a implementation of Focal Loss with smooth label cross entropy supported which is proposed in
1010
'Focal Loss for Dense Object Detection. (https://arxiv.org/abs/1708.02002)'
1111
Focal_Loss= -1*alpha*(1-pt)*log(pt)
12-
:param num_class:
1312
:param alpha: (tensor) 3D or 4D the scalar factor for this criterion
1413
:param gamma: (float,double) gamma > 0 reduces the relative loss for well-classified examples (p>0.5) putting more
1514
focus on hard misclassified example

0 commit comments

Comments
 (0)