torch>=0.4.1
torchvision>=0.2.2
scikit-learn>=0.23.0
