matplotlib==3.0.2 numpy==1.15.4 scipy==1.1.0 sklearn==0.0 torch==0.4.1 torchvision==0.2.1