updates

2019-03-04 16:06:13 +01:00 · 2019-03-04 16:06:13 +01:00 · 5fcdcefec3
parent 5ca987eaed
commit 5fcdcefec3
1 changed files with 1 additions and 0 deletions
--- 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])