torch
torchvision
argparse
PIL
matplotlib
