Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/ikeasamoahansah/vit

Pytorch implementation of the Vision Transformer Model
https://github.com/ikeasamoahansah/vit

Last synced: 1 day ago
JSON representation

Pytorch implementation of the Vision Transformer Model

Awesome Lists containing this project

README

        

ViT


Pytorch implementation of the Vision Transformer Model

### Steps to reproduce

1. Clone and install dependencies
```
!git clone https://github.com/ikeasamoahansah/ViT
%cd ViT
!pip install -r requirements.txt
```

2. Start training
```
!python src/train.py
```

3. Voila