torch==1.4.0
torchvision==0.5.0
numpy
scikit-learn==0.22.2
pillow==7.0
