vision-transformer-cover

这是一个vit的复现库，包含从模型结构到训练框架的一系列脚本。由pytorch实现 本训练使用cifar-10训练集实现。 使用方法： 1.下载源码 2.运行train脚本，自动下载cifar-10数据集。 3.运行main文件，开始训练。训练的模型参数将会被保存在weight文件夹中。