torch
torchvision
matplotlib
seaborn
numpy