torchvision
torch
scikit-learn
tqdm
numpy
matplotlib
