numpy
torchvision
torch
tqdm