diff --git a/nets/yolo_training.py b/nets/yolo_training.py index b5eede114219b7325151386810f648669b5d8e4b..da2f1f9abe2b88f9b54207ad08249bbed945bf76 100644 --- a/nets/yolo_training.py +++ b/nets/yolo_training.py @@ -220,7 +220,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