updates

2019-02-21 10:57:55 +01:00
parent 7a31f4b288
commit 646a21a5cd
1 changed files with 2 additions and 1 deletions
@@ -159,7 +159,8 @@ class YOLOLayer(nn.Module):
            # Compute losses
            nT = sum([len(x) for x in targets])  # number of targets
            nM = mask.sum().float()  # number of anchors (assigned to targets)
-            k = nM / bs
+            k = 1 #nM / bs
+
            if nM > 0:
                lxy = k * MSELoss(xy[mask], txy[mask])
                lwh = k * MSELoss(wh[mask], twh[mask])