torch
torchvison
numpy
