chongruo / detectron2-ResNeSt

A fork of Detectron2 with ResNeSt backbone

Home Page:https://arxiv.org/abs/2004.08955

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

exnx opened this issue · comments

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