[PPYOLOE] fix loss bug in negative batch (#5775)

d63faf03 · shangliang Xu · GitHub · c575dea3 · d63faf03
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/heads/ppyoloe_head.py ppdet/modeling/heads/ppyoloe_head.py +1 -1

未找到文件。
--- a/ppdet/modeling/heads/ppyoloe_head.py
+++ b/ppdet/modeling/heads/ppyoloe_head.py
@@ -293,7 +293,7 @@ class PPYOLOEHead(nn.Layer):
        else:
            loss_l1 = paddle.zeros([1])
            loss_iou = paddle.zeros([1])
-            loss_dfl = paddle.zeros([1])
+            loss_dfl = pred_dist.sum() * 0.
        return loss_l1, loss_iou, loss_dfl

    def get_loss(self, head_outs, gt_meta):