search

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

Is CPU inference possible? #63

Open scalavia87 opened 1 year ago

scalavia87 commented 1 year ago

Is possible to use a detectron2-resnest trained model to infere using only CPU? I tried changing DEVICE from cuda to CPU in config.yaml file, but i get a NotImplemented error. (This works in original Detectron2)

Thanks in advance.