使用argparse在终端运行PyTorch模型