search

davidnvq / grit

GRIT: Faster and Better Image-captioning Transformer (ECCV 2022)
185 stars 28 forks source link

Replacement for MSDeformAttn? #7

Closed shayanjoya closed 2 years ago

shayanjoya commented 2 years ago

Hi, thanks for your work! is there a drop is replacement in pure pytorch for MSDeformAttn which we can use instead, or an alternative you can recommend? since it is not implemented for CPU usage.

davidnvq commented 2 years ago

Thanks for your interest. I haven't tried the CPU-based implementation but I found a similar issue, which points out the author's implementation for debugging: https://github.com/fundamentalvision/Deformable-DETR/issues/131 Should you try it.