torch
torchvision
torchsummary
numpy
