sklearn
torch
torchvision
tqdm
numpy
matplotlib
