python的argparse模块使用笔记(1)

本文介绍了Python的argparse模块中的parse_known_args()方法,该方法在处理命令行参数时允许保留未知参数,以便后续处理。通过示例展示了如何在深度学习代码中使用此功能,先处理基本选项,然后逐步添加更多参数而不会引发错误。在未定义的选项被解析时,argparse会保存这些参数,直到最终解析所有选项。此外,还提供了一个错误示例,说明当传递未配置的选项时,程序将在最后报错。
摘要由CSDN通过智能技术生成

argparse是一个非常方便和常见的模块,在深度学习的代码实现中用的非常多,但是平时用的时候也会遇到很多问题,记录一下,不断补充

2021.7.7

遇到了parse_known_args() , 找了一篇写的非常好的博客,搬过来作为自己的学习笔记,怕后面找不到了。

parse_known_args()方法的用处就是有些时候,你的选项配置可能不会在一个函数或包中调用完成。例如在很多使用,我们可能会需要根据一些输入的选项。

在深度学习中,我们可能会根据传入的模型设置–model来决定我们调用的是哪个模型代码,然后在该模型中还会有一部分的选项设置。

那么这时候就会出现一种情况就是运行命令中会传入所有需要设置的选项值,但是有时候仅获取到基本设置时可能要进行一些操作然后才继续导入剩下的参数设置。分开导入又太麻烦,此时使用parse_known_args()是一个非常好的选择。

parse_known_args()方法的作用就是当仅获取到基本设置时,如果运行命令中传入了之后才会获取到的其他配置信息,先不会报错,并将多出来的部分保存起来,留到后面使用。

import argparse
def basic_options():
    parser = argparse.ArgumentParser()
    parser.add_argument('--data_mode', type=str, default= 'unaligned', help='chooses how datasets are loaded')
    parser.add_argument('--mode', type=str, default='test', help='test mode')
    return parser

def data_options(parser):
    parser.add_argument('--lr', type=str, default='0.0001', help='learning rate')
    return parser

if __name__ == '__main__':
    parser = basic_options()
    opt, unparsed = parser.parse_known_args()
    print(opt)
    print(unparsed)
    parser = data_options(parser)
    opt = parser.parse_args()
    print(opt)

运行结果:

(deeplearning) me@computer:~$ python test_data.py --data_mode aligned --lr 0.0002
Namespace(data_mode='aligned', mode='test')
['--lr', '0.0002']
Namespace(data_mode='aligned', lr='0.0002', mode='test')

该例子说明了在一开始仅导入了basic_options()选项时,多余出来的–lr选项会被保存起来,不会报错,直到接下来导入了data_options(parser)之后再将其赋值

这时候如果我们传入一个没有配置的选项,它在中间的时候也会保存起来,但是最后就会报错:

(deeplearning) me@computer:~$ python test_data.py --data_mode aligned --lr 0.0002 --no_test True
Namespace(data_mode='aligned', mode='test')
['--lr', '0.0002', '--no_clue', 'True']
usage: test_data.py [-h] [--data_mode DATA_MODE] [--mode MODE] [--lr LR]
test_data.py: error: unrecognized arguments: --no_test True

其他:如果想要以字符串形式获取选项值,在最后添加一行:

print('{data_mode}_{mode}_{lr}'.format(**vars(opt)))

返回:

(deeplearning) me@computer:~$ python test_data.py --data_mode aligned --lr 0.0002
Namespace(data_mode='aligned', mode='test')
['--lr', '0.0002']
Namespace(data_mode='aligned', lr='0.0002', mode='test')
aligned_test_0.0002

参考:

https://www.cnblogs.com/wanghui-garcia/p/11267160.html

未完待续

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值