numpy>=1.18.5
torchvision>=0.9.1
torch>=1.8.1
pytorch_lightning>=1.2.5
vit-pytorch>=0.6.7
lightning-bolts>=0.3.2
