search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.1k stars 230 forks source link

Inference on CPU? #190

Open htcml opened 1 year ago

htcml commented 1 year ago

Can DINO models be inferenced on CPU? The inference notebook always looks for cuda.

HaoZhang534 commented 1 year ago

It temporally only support GPU inference.