torch
torchvision
tqdm