torch
torchvision
tqdm
matplotlib
numpy
