inference of detectron2 on CPU

[SylvainArd]

Hello, I have trained my data on GPU with maskrcnn and ResNet 101 and want to deploy it on CPU by demo.py. The code is written for GPU, how can I change it to works on CPU please ? Thank you !
Best regards

[sukjunhwang]

Adding MODEL.DEVICE cpu at the end of your command line will change the device to cpu.
You can check this at

detectron2/detectron2/config/defaults.py

Line 32 in 2a1cec4

_C.MODEL.DEVICE = "cuda"