How can I use the demo (inference) on cpu?

chongruo / detectron2-ResNeSt

A fork of Detectron2 with ResNeSt backbone

https://arxiv.org/abs/2004.08955

Apache License 2.0

385 stars 73 forks source link

How can I use the demo (inference) on cpu? #52

Open exnx opened 3 years ago

exnx commented 3 years ago

Hi, is there a way I can use the demo.py on cpu? When I run it and it tries to load, it breaks because the pretrained model is for gpu. I couldn't find the option / flag to turn off cuda. Is that possible for the demo script? Thanks!

Eric