Skip to content

Commit

Permalink
cpu model
Browse files Browse the repository at this point in the history
  • Loading branch information
GOATmessi8 committed Dec 25, 2017
1 parent a1033fa commit c86425a
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions test_RFB.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,8 @@

priorbox = PriorBox(cfg)
priors = Variable(priorbox.forward(), volatile=True)
if not args.cuda:
prior = priors.cpu()


def test_net(save_folder, net, detector, cuda, testset, transform, max_per_image=300, thresh=0.005):
Expand Down Expand Up @@ -175,6 +177,8 @@ def test_net(save_folder, net, detector, cuda, testset, transform, max_per_image
if args.cuda:
net = net.cuda()
cudnn.benchmark = True
else:
net = net.cpu()
# evaluation
#top_k = (300, 200)[args.dataset == 'COCO']
top_k = 200
Expand Down

0 comments on commit c86425a

Please sign in to comment.