torch>=1.13
torchvision>=0.14
numpy
gdown
