torch
torchvision
pyyaml
numpy