torch
torchvision
numpy
PyYAML
