From 4821507c859b97841175eeb300040e1598420e5f Mon Sep 17 00:00:00 2001
From: Bubbliiiing <3323290568@qq.com>
Date: Sun, 10 Jul 2022 11:42:45 +0800
Subject: [PATCH] update loss

---
 nets/yolo_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nets/yolo_training.py b/nets/yolo_training.py
index eac5964..484ea20 100644
--- a/nets/yolo_training.py
+++ b/nets/yolo_training.py
@@ -177,7 +177,7 @@ class YOLOLoss(nn.Module):
         bs          = tobj.shape[0]
         
         loss    = box_loss + obj_loss + cls_loss
-        return loss * bs
+        return loss
         
     def xywh2xyxy(self, x):
         # Convert nx4 boxes from [x, y, w, h] to [x1, y1, x2, y2]
-- 
GitLab