torch>=1.2
torchvision
numpy
pillow
scikit-learn
matplotlib
tqdm
