scipy==1.2.1
numpy==1.14.5
torch==1.7.0
torchvision==0.8.0
sklearn