xmu-xiaoma666

External-Attention-pytorch

xmu-xiaoma666

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

AI 简介

这是一个面向深度学习研究者的轻量级模块化代码库,提供多种注意力机制(如External Attention、CBAM、MobileViTv2Attention等)、MLP变体、重参数化方法及卷积改进模块的PyTorch独立实现。所有模块均剥离任务框架依赖,以简洁、可复用的类形式封装,便于快速理解论文核心思想与验证新设计。适用于计算机视觉方向的论文复现、模型结构探索、教学演示及工业级轻量化模型搭建。

Python
MIT License
12.2k
Stars
1.9k
Forks
102
Watchers
66
Issues

Star 增长

今日0
近 7 天0
近 30 天+3
综合评分70.17
默认分支master