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)