numpy
scikit-learntqdm
torch
torchvision
