search

hkchengrex / Tracking-Anything-with-DEVA

[ICCV 2023] Tracking Anything with Decoupled Video Segmentation
https://hkchengrex.com/Tracking-Anything-with-DEVA/
Other
1.27k stars 129 forks source link

Deva on CPU #61

Closed riccardo24-7 closed 9 months ago

riccardo24-7 commented 9 months ago

Hi, How can I deploy the deva project on CPU. I have only errors connected with

ERROR Got unsupported ScalarType BFloat16 or ERROR Input type (torch.cuda.HalfTensor) and weight type (torch.FloatTensor) should be the same

Thanks

hkchengrex commented 9 months ago

Hello. We did not write the code with non-CUDA companionability in mind. It can be fixed by removing the autocast and the .cuda() calls.