updates

2019-02-20 12:52:39 +01:00 · 2019-02-20 12:52:39 +01:00 · f728bd21d2
parent 315fe6ec14
commit f728bd21d2
1 changed files with 0 additions and 1 deletions
--- 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)