torch
numpy
torchvision
scipy
