Unverified Commit 454bc14d authored by Sai-Suraj-27's avatar Sai-Suraj-27 Committed by GitHub
Browse files

fix: Removed a wrong key-word argument in `sigmoid_focal_loss()` function call (#31951)

Removed a wrong key-word argument in sigmoid_focal_loss() function call.
parent a5c642fe
......@@ -2163,7 +2163,7 @@ class RTDetrLoss(nn.Module):
target_classes[idx] = target_classes_original
target = F.one_hot(target_classes, num_classes=self.num_classes + 1)[..., :-1]
loss = sigmoid_focal_loss(src_logits, target, self.alpha, self.gamma, reduction="none")
loss = sigmoid_focal_loss(src_logits, target, self.alpha, self.gamma)
loss = loss.mean(1).sum() * src_logits.shape[1] / num_boxes
return {"loss_focal": loss}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment