timm的vit (vision transformer), swin Transformer源码仓库-无优化版

目录

1.  VIT

1.1 pytorch自带版

1.2 基于timm实现

2. Swin transformer

---

1.  VIT

1.1 pytorch自带版

https://github.com/lucidrains/vit-pytorch

1.2 基于timm实现

放一个github上的无优化版仓库，可以基于这个来魔改代码：

https://github.com/UCSC-VLAA/FedConv/blob/10abdb11123420915c97f1e7d37b08eba71f0eb1/models/vision_transformers.py#L21

(上面那个可能更加官方)

mm-cot/timm/models/vision_transformer.py at 8dd4ac02b94f21347973491f6e6b828502d23f9d · amazon-science/mm-cot · GitHub

2. Swin transformer

swin transformer的官方代码：

Swin-Transformer/main.py at main · microsoft/Swin-Transformer · GitHub

swin transformer的讲解可以看这里：

notes/swin_transformer.md at 7f7913d50524c0bf7f4c37c379a444384360bbb3 · NairongZheng/notes · GitHub