yacs
torchvision
numpy