mitmul / chainer-faster-rcnn

Object Detection with Faster R-CNN in Chainer
MIT License
288 stars 87 forks source link

bug fixed forward.py errors in cpu environment #2

Closed shiba24 closed 8 years ago

shiba24 commented 8 years ago

Users who don't have no GPUs cannot run "import Device". So I changed codes to run cuda.Device if args.gpu >= 0.

mitmul commented 8 years ago

@shiba24 Great, thanks!