torch
torchvision
numpy
pillow
scikit-learn
tqdm
matplotlib
visdom