search

IDEA-Research / Stable-DINO

[ICCV 2023] Official implementation of the paper "Detection Transformer with Stable Matching"
https://arxiv.org/abs/2304.04742
Apache License 2.0
218 stars 7 forks source link

Inference on CPU #1

Open htcml opened 1 year ago

htcml commented 1 year ago

Can Stable-DINO do inference on CPU? DINO only can do inference on GPU due to using deformable attention, so wonder if Stable-DINO also has the same issue.

rentainhe commented 1 year ago

There's CPU version of Deformable-Attention, so both DINO and Stable-DINO can do inference on CPU