numpy
scikit-learn
tqdm
torch
torchvision
