PyTorch-MNIST-Tutorial/ ├── LICENSE ├── model.pth ├── model.py ├── README.md ├── test.py ├── train.py ├── utils.py ├── demo.py ├── mnist_train.csv └── mnist_test.csv ...
Pytorch reimplementation of Google's repository for the ViT model that was released with the paper An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale by Alexey Dosovitskiy, ...