torch
torchvision
pyyaml