本项目仅为学习项目,仅供参考,并不做实际应用
- img_for_md:-> 用于markdown的图片
- FormationForResnet18.png -> Resnet18结构图
- PartOfViT: -> ViT中的各个部件,练习用
- attention.py -> 注意力机制
- transformers -> 阉割版ViT,不包含<注意力机制,class token ,position_embedding>
- Resnet18: -> 复现Resnet,有部分修改
- Resnet -> 定义网络结构
- dataset -> 加载数据集
- main_signal.py -> 主函数 训练 & 测试
- utils -> 统计
- ViT完整版:
- ViT -> 完整复现ViT
- 2010.11929.pdf -> 原论文
- img.jpg -> 从MNIST数据集中的一张图片
- test.py -> 搞不懂的时候测试用的