torch>=1.6.0
torchvision
tqdm
scipy
scikit-learn
