Where is the Neighborhood Attention 2D?
ady-yu opened this issue · comments
ady-yu commented
I wonder where is the Neighborhood Attention 2D. I can't find the code
Steven Walton commented
Our standard NA is by default 2D, so it should be in the natten folder. Unless otherwise specified, NA is 2D.
Ali Hassani commented
Fixed in PR #55.