Merge pull request facebookresearch#1029 from Dorozhko-Anton/1013-sig…

…moid_focal_loss_gamma_alpha gamma, alpha are supposed to be floats, delete lines with [0] subscripts facebookresearch#1013
etarakci-hvl · Sep 16, 2019 · 03a5949 · 03a5949
2 parents 4d19749 + a393875
commit 03a5949
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/maskrcnn_benchmark/layers/sigmoid_focal_loss.py b/maskrcnn_benchmark/layers/sigmoid_focal_loss.py
@@ -39,8 +39,6 @@ def backward(ctx, d_loss):
 
 def sigmoid_focal_loss_cpu(logits, targets, gamma, alpha):
     num_classes = logits.shape[1]
-    gamma = gamma[0]
-    alpha = alpha[0]
     dtype = targets.dtype
     device = targets.device
     class_range = torch.arange(1, num_classes+1, dtype=dtype, device=device).unsqueeze(0)