torch
torchvision
matplotlib