torch==1.3.1
torchvision==0.4.2
numpy==1.18.1
scikit-learn==0.22.2
pillow==6.2.2
