updates

2019-03-04 16:01:23 +01:00 · 2019-03-04 16:01:23 +01:00 · 5ca987eaed
parent 175e231c55
commit 5ca987eaed
1 changed files with 2 additions and 2 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -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

    for b in range(nB):
-        t = target[b]
+        t = target[b].cpu()
        nTb = len(t)  # number of targets
        if nTb == 0:
            continue