ArgumentParser
ArgumentParser
更具体细节的讲解可以参考知乎_Link
创建parser
1 | import argparse |
创建parser对象,可以传入一个字符串作为描述信息。
添加参数
1 | parser.add_argument("--max_train_steps", type=int, default=int(3e6), help=" Maximum number of training steps") |
--max_train_steps
:参数名,前面加--
表示是可选参数type=int
:参数类型default=int(3e6)
:默认值help=" Maximum number of training steps"
:帮助信息, 用于-h
时显示
解析命令行参数
1 | args = parser.parse_args() |
得到args对象,包含了所以之前的参数,之后都过传递args对象来调用定好的参数.
All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.