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:55:06 +01:00
parent 344bea20eb
commit ed37551c38
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -158,7 +158,7 @@ class YOLOLayer(nn.Module):
tcls = tcls[mask]
if xy.is_cuda:
txy, twth, mask, tcls = txy.cuda(), twh.cuda(), mask.cuda(), tcls.cuda()
txy, twh, mask, tcls = txy.cuda(), twh.cuda(), mask.cuda(), tcls.cuda()
# Compute losses
nT = sum([len(x) for x in targets]) # number of targets