torch
torchvision
tqdm
six
statsmodels
torchsummary
matplotlib
numpy