argparse是python用于解析命令行参数和选项的标准模块,一方面可以接收运行代码时用户传入的各种参数(比如传入数据路径),另一方面也可以规范的定义程序中用到的参数(比如定义epoch及batch_size等)。
①导入包:import argparse
②创建解析器:parser = argparse.ArgumentParser()
③添加参数:parser.add_argument('--epochs', type=int, default=100, help='')
④解析参数:args = parser.parse_args()
⑤使用参数:epoch = args.epochs
import argparse # 导入包
parser = argparse.ArgumentParser() # 创建解析器
# 添加参数
parser.add_argument('--test', action='store_true', default=False, help='Test module.')
parser.add_argument('--batch_size', type=int, default=32, help='Total batch size.')
parser.add_argument('--epochs', type=int, default=100, help='Num of epochs to train.')
args = parser.parse_args() # 解析参数
# 使用
print(args.epochs)
100
print(args.batch_size)
32
python test.py --batch_size 64
print(args.test)
False
print(args.batch_size)
64
print(args.epochs)
100
python test.py --test --epochs 50
print(args.test)
True
print(args.batch_size)
32
print(args.epochs)
50
action=‘store_true’ ,只要运行时该变量有传参,该变量就为True。