torch==0.4.1
torchvision
tqdm
