matplotlib
numpy
torch
pandas
numpy
torchsummary
thop
pygame
