search

ucbdrive / few-shot-object-detection

Implementations of few-shot object detection benchmarks
Apache License 2.0
1.08k stars 225 forks source link

running it on CPU? #112

Closed PradeepThapa closed 2 years ago

PradeepThapa commented 3 years ago

Hi,

How can I run run_experiments.py in CPU?

xinw1012 commented 3 years ago

Hi, sorry for the delayed response. Our current configurations and results were obtained on multi-GPU settings. Since our codebase is built upon detectron2, if you could run the vanilla detectron2 on CPU only, you should be able to use or adapt our current codebase. I found the prebuilt detectron2 on CPU here https://detectron2.readthedocs.io/en/latest/tutorials/install.html so you might try to install the corresponding detectron2 version. We would appreciate it if you could keep this thread open and let us know if the current codebase is compatible with the CPU-only version. We would be happy to keep it compatible. PR would be highly appreciated!

Thanks for your interest in our work!

slgxmh commented 2 years ago

Add this config in your config yaml.

MODEL:
  DEVICE: "cpu"