thuml/MobileAttention

Official implementation of "Mobile Attention: Mobile-Friendly Linear-Attention for Vision Transformers in PyTorch". To run the code, you can refer to https://github.com/thuml/Flowformer.

语言：Python · GitHub 星标：9 · Fork：0

在 GitHub 上查看原仓库