torch>=1.1.0
torchvision>=0.2.1
scikit-learn>=0.19.2
pandas
