search

MasterBin-IIAU / Unicorn

[ECCV'22 Oral] Towards Grand Unification of Object Tracking
MIT License
949 stars 87 forks source link

CPU-mode not supported? #25

Closed carsten12 closed 2 years ago

carsten12 commented 2 years ago

Under the assumption that I can try out Unicorn with some pre-trained models, I tried to install it on a Mac and a PC, both w/o nVidia GPU. In step 3 when executing bash make.sh, setup.py is called which is testing for torch.cuda.is_available() and is raising an exception if the function returns false. So running in CPU-mode is not foreseen?

MasterBin-IIAU commented 2 years ago

Hi, Unicorn uses Deformable Attention for feature interaction. Unfortunately, this operation is not supported to run on the CPU for now.