bug fix

a2079427 · Kai Chen · 13dec7ea · a2079427
Commit a2079427 authored Nov 26, 2018 by Kai Chen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/core/loss/losses.py mmdet/core/loss/losses.py +1 -1

No files found.
--- a/mmdet/core/loss/losses.py
+++ b/mmdet/core/loss/losses.py
@@ -31,7 +31,7 @@ def sigmoid_focal_loss(pred,
                       gamma=2.0,
                       alpha=0.25,
                       reduction='elementwise_mean'):
-    pred_sigmoid = pred.sigmoid().detach()
+    pred_sigmoid = pred.sigmoid()
    pt = (1 - pred_sigmoid) * target + pred_sigmoid * (1 - target)
    weight = (alpha * target + (1 - alpha) * (1 - target)) * weight
    weight = weight * pt.pow(gamma)