From 5fcdcefec343520dd42988cb6f0f7a75859bea42 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 4 Mar 2019 16:06:13 +0100 Subject: [PATCH] updates --- models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models.py b/models.py index 1ecfdc18..f3d57e68 100755 --- a/models.py +++ b/models.py @@ -161,6 +161,7 @@ class YOLOLayer(nn.Module): nM = mask.sum().float() # number of anchors (assigned to targets) k = 1 # nM / bs if nM > 0: + print(xy.shape, txy.shape, mask.shape) lxy = k * MSELoss(xy[mask], txy[mask]) lwh = k * MSELoss(wh[mask], twh[mask])