torch>=0.4
torchvision
tqdm
