search

SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
MIT License
1.04k stars 85 forks source link

Where is the Neighborhood Attention 2D? #49

Closed ady-yu closed 1 year ago

ady-yu commented 2 years ago

I wonder where is the Neighborhood Attention 2D. I can't find the code

stevenwalton commented 2 years ago

Our standard NA is by default 2D, so it should be in the natten folder. Unless otherwise specified, NA is 2D.

alihassanijr commented 1 year ago

Fixed in PR #55.