torch==1.7.0
torchvision==0.8.1
numpy
tqdm
matplotlib
dotmap
Pillow