Update yolo_training.py

daded78f · Bubbliiiing · GitHub · 27cb95f1 · daded78f
隐藏空白更改
内联并排

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
@@ -194,7 +194,7 @@ class YOLOLoss(nn.Module):
            num_pos = torch.sum(mask)
            num_pos = torch.max(num_pos, torch.ones_like(num_pos))
        else:
-            num_pos = bs
+            num_pos = bs/3
        return loss, num_pos
    def get_target(self, target, anchors, in_w, in_h, ignore_threshold):