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