From 0b3a17362cccf95a92a7e93a0d8fa8bae660802b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 4 Mar 2019 15:56:58 +0100
Subject: [PATCH] updates

---
 models.py | 2 ++
 1 file changed, 2 insertions(+)

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]