search

hkchengrex / Cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation
https://hkchengrex.com/Cutie/
MIT License
579 stars 60 forks source link

Use specified device type to use CPU only #79

Closed ashishp03 closed 4 days ago

ashishp03 commented 2 weeks ago

What is the way to specify the device type, for example, if I want to specifically use the CPU, how can I change the code to fit this?

hkchengrex commented 1 week ago

We officially only support CUDA. I believe it should not be too hard to hack it to work on CPU, e.g., by changing https://github.com/hkchengrex/Cutie/blob/e869fab5e653b118c475cbafe0b2322b37a7e320/cutie/utils/get_default_model.py#L23

and other .cuda() calls in the scripting demo.

hkchengrex commented 4 days ago

Please feel free to re-open if there are any updates.