torch>=1.4.0
torchvision>=0.5.0
scikit-learn==0.23.2
scipy==1.5.3
