DeepSolo icon indicating copy to clipboard operation
DeepSolo copied to clipboard

Published 20 hours ago • verified publisher icon ViTAE-Transformer

Implement on CPU

Open rwood-97 opened this issue 10 months ago • 1 comments

This PR implements DeepSolo on CPU by calling the pytorch implementation of multiscale deformable attention if cuda is not found. I have also organised the cpu/cuda files into cpu/cuda folders respectively just for clarity.

rwood-97 avatar Apr 09 '24 13:04 rwood-97

Fixes #22 as well

rwood-97 avatar Apr 09 '24 13:04 rwood-97