From 3f21d5bb2e71fde69b0b288514db80f0e9505e33 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 4 Mar 2019 16:13:31 +0100 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index e78a3e62..c214fd2d 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -242,7 +242,7 @@ def build_targets(target, anchor_vec, nA, nC, nG): tconf = torch.ByteTensor(nB, nA, nG, nG).fill_(0) tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes - if anchor_vec.is_cuda(): + if anchor_vec.is_cuda: txy = txy.cuda() twh = twh.cuda() tconf = tconf.cuda()