torchvision
torch
numpy
tqdm
