Pytorch基本用法 argparse模块详解：处理命令行参数

1、介绍

argparse是py.3标准库中用来处理命令行参数的模块，用于提供命令行接口（本文仅windows）。

2、使用步骤

创建ArgumentParser()对象

调用add_argument()方法添加参数

使用parse_args()解析添加的参数

3、实例

（1）创建py.文件（test.py）。

（2）开始在输入模式(i)下输入code。

import argparse

parser = argparse.ArgumentParser() #创建一个对象

parser.add_argument("square", help = "pls enter a num", type = int) #添加一个叫square的参数，类型必须是int

parser.add_argument("input_str", help = "pls enter a string", type = str) #添加一个叫input_str的参数，类型必须是str

parser.add_argument('--eval', action="store_true", help = "Whether to perform evaluation after training") #添加一个叫eval的参数

parser.add_argument('--alpha', type = float, default = 0.02, help = "Weighting hyperparameter for loss function") #添加一个叫alpha的参数，用于某个任务（计算loss）

args = parser.parse_args() #解析

print(args.square**2)

print(args.input_str)

print(args.eval)

print(args.alpha)

（3）win+R进入命令行，cd至路径下，运行下面的程序

python test.py 1 "hello string!" --eval --alpha 0.02

（4）可以运行下面程序，查看文件的参数信息

python test.py -h

（5）运行结果如下图 

  4、另外

（1） 如上图，square、input_str是positional参数，eval、alpha是设置的可选参数，如果要设置必选参数(使用required)，可以按照下面代码

parser.add_argument('--num', type=int, required=True, help="must enter a num")

（2）另外参数可以是文件

parser.add_argument('--file', type=argparser.FileType('r')) # 读取文件
parser.add_argument('--file', default='text.txt', type=argparser.FileType('r')) # 读取文件，预定义参数