pyyaml
ipdb
scikit-learn
numpy
torch
setuptools
torch>=1.2.0
torchvision>=0.4.0
matplotlib
pillow
