torch
numpy
torchvision
argparse
Pillow
sklearn
pandas
seaborn
scipy
matplotlib
tensorboardX
