From ec308d605e66b586db10754550e0705d79a303d0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 21 Feb 2019 10:59:05 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index e6832ab5..2386b222 100755 --- 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])