zzd1992 / Image-Local-Attention

A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.
136Updated 2 years ago

Related projects

Alternatives and complementary repositories for Image-Local-Attention