apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-03-04 16:01:23 +01:00
parent 175e231c55
commit 5ca987eaed
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -243,7 +243,7 @@ def build_targets(target, anchor_vec, nA, nC, nG):
tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes
for b in range(nB): for b in range(nB):
t = target[b] t = target[b].cpu()
nTb = len(t) # number of targets nTb = len(t) # number of targets
if nTb == 0: if nTb == 0:
continue continue