Skip to content

Commit 14ddd12

Browse files
committed
fix yolov7 loss amp
1 parent 3a60a87 commit 14ddd12

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ppdet/modeling/losses/yolov7_loss.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,8 @@ def build_targets(self, p, targets, anchors, batch_images):
349349

350350
y = cls_preds_.sqrt_()
351351
pairwise_cls_loss = F.binary_cross_entropy_with_logits(
352-
paddle.log(y / (1 - y)), gt_cls_per_image,
352+
paddle.log(y / (1 - y) + 1e-8),
353+
gt_cls_per_image,
353354
reduction="none").sum(-1)
354355
del cls_preds_
355356

@@ -543,7 +544,8 @@ def build_targets2(self, p, targets, anchors, batch_images):
543544

544545
y = cls_preds_.sqrt_()
545546
pairwise_cls_loss = F.binary_cross_entropy_with_logits(
546-
paddle.log(y / (1 - y)), gt_cls_per_image,
547+
paddle.log(y / (1 - y) + 1e-8),
548+
gt_cls_per_image,
547549
reduction="none").sum(-1)
548550
del cls_preds_
549551

0 commit comments

Comments
 (0)