Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-20 12:52:39 +01:00
parent 315fe6ec14
commit f728bd21d2
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -130,7 +130,6 @@ class YOLOLayer(nn.Module):
if p.is_cuda: if p.is_cuda:
self.grid_xy = self.grid_xy.cuda() self.grid_xy = self.grid_xy.cuda()
self.anchor_vec = self.anchor_vec.cuda()
self.anchor_wh = self.anchor_wh.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) # p.view(bs, 255, 13, 13) -- > (bs, 3, 13, 13, 80) # (bs, anchors, grid, grid, classes + xywh)