torch==1.8.1
torchvision
Pillow
tqdm