tjyuyao/Image-Local-Attention
A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.
Python
Watchers
No one’s watching this repository yet.
A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.
Python
No one’s watching this repository yet.