NeighborhoodAttention1d

SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022

MIT License

1.05k stars 86 forks source link

Closed alihassanijr closed 2 years ago

alihassanijr commented 2 years ago

Adds a 1D CUDA kernel (#30)
Refactors the extension by moving it to root, and creating symlinks for the three tasks.
Fixes the missed refactor from the last PR in DET and SEG's flop counters.