diff --git a/models.py b/models.py
index 1ecfdc18..a323e87e 100755
--- a/models.py
+++ b/models.py
@@ -150,6 +150,8 @@ class YOLOLayer(nn.Module):
             p_conf = p[..., 4]  # Conf
             p_cls = p[..., 5:]  # Class
 
+            print(self.anchor_vec.device, self.anchor_vec, self.nA, self.nC, nG)
+
             txy, twh, mask, tcls = build_targets(targets, self.anchor_vec, self.nA, self.nC, nG)
 
             tcls = tcls[mask]