torch>=1.6.0
numpy
scikit-learn
pyyaml
pillow