torch
matplotlib
numpy
tqdm
torchvision
scipy
