Pytorch ViT for Image classification on the CIFAR10 dataset

Introduction

This project uses ViT to perform image classification tasks on DATA set CIFAR10. The implement of Vit and pretrained weight are from https://github.com/asyml/vision-transformer-pytorch.

Create environment:

conda create --name vit --file requirements.txt
conda activate vit

Download the CIFAR10 from http://www.cs.toronto.edu/~kriz/cifar.html, creat data floder and unzip the cifar-10-python.tar.gz in ‘data/’.

python main.py