torch
torchvision
argparse
matplotlib
