demo ssd in only cpu inference but no gpu

[kuonangzhe]

Thanks for the awesome work!

I just tested the demo_ssd.py on my laptop. I found that the args.gpu for setting ctx was no used in the code actually, and I want to test the gpu speed, so I set '--gpu 0' in args, and get model like follows:

net = gcv.model_zoo.get_model(args.network, pretrained=False, ctx=ctx, pretrained_base=False)
net.load_params(args.pretrained)

The weird thing is that the speed for ssd has no change. In my case, on a GTX 940M, with ssd_512_coco model, the speed for cpu and gpu are both around 2.7 seconds. The time is in average time.

Is there any place that I can check for this? Thanks in advance!

[Angzz]

First,the param is —gpus not gpu, please have a check, second, maybe the gpu you use is not powerful, I think the 940 M is too slow to inference.

[zhreshold]

Fixed in #391

[kuonangzhe]

@zhreshold
Thanks a lot!