学习之余验证复现Vision transformer
# 在仓库中自行下载自定义数据集,运行对应jupyter notebook即可复现
# Vit实际从头训练需要消耗极大算力,因此同时提供了迁移学习版本
(base) nico@GAME3070:~/Vit$ tree data/ -d
data/
└── pizza_steak_sushi
├── test
│ ├── pizza
│ ├── steak
│ └── sushi
└── train
├── pizza
├── steak
└── sushi