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