updates

2019-02-21 10:59:05 +01:00 · 2019-02-21 10:59:05 +01:00 · ec308d605e
parent 646a21a5cd
commit ec308d605e
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -159,7 +159,7 @@ 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 = 1 #nM / bs
+            k = 1  # nM / bs

            if nM > 0:
                lxy = k * MSELoss(xy[mask], txy[mask])