update loss

4821507c · Bubbliiiing · 53c50566 · 4821507c
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

nets/yolo_training.py nets/yolo_training.py +1 -1

未找到文件。
--- 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]