torch>=1.5.0
matplotlib
numpy
tqdm
torchvision
scipy
