We develop NATTEN (Neighborhood Attention Extension), a Python package | |
with efficient C++ and CUDA kernels, which allows NA to run up to 40% faster than Swin's WSA while using up to 25% less | |
memory. |
We develop NATTEN (Neighborhood Attention Extension), a Python package | |
with efficient C++ and CUDA kernels, which allows NA to run up to 40% faster than Swin's WSA while using up to 25% less | |
memory. |