numpy
torch>=1.7.0
torchvision>=0.8.0
scikit-learn
pillow
