From f728bd21d23b9e44f6458d771601370374591225 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 20 Feb 2019 12:52:39 +0100
Subject: [PATCH] updates

---
 models.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/models.py b/models.py
index 3a20fcce..8beff489 100755
--- a/models.py
+++ b/models.py
@@ -130,7 +130,6 @@ class YOLOLayer(nn.Module):
 
                 if p.is_cuda:
                     self.grid_xy = self.grid_xy.cuda()
-                    self.anchor_vec = self.anchor_vec.cuda()
                     self.anchor_wh = self.anchor_wh.cuda()
 
         # p.view(bs, 255, 13, 13) -- > (bs, 3, 13, 13, 80)  # (bs, anchors, grid, grid, classes + xywh)