torch
torchvision
numpy
gdown
