gdown
numpy
matplotlib
Pillow
torch>1.12
torchvision