A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool