torch==1.3.0 torchvision>=0.4 numpy==1.17.2 matplotlib==3.1.1 scipy==1.3.1