argparse
是 Python 标准库中用于解析命令行参数的模块。它允许编写用户友好的命令行界面,并方便地处理命令行参数和选项。
# train.py
parser = argparse.ArgumentParser(description='Train an editor')
parser.add_argument('config', help='train config file path')
parser.add_argument('--work-dir', help='the dir to save logs and models')
argparse 允许定义脚本应该接受哪些命令行参数。这些参数可以是位置参数(按照出现的顺序解析)或可选参数(使用 – 或 - 前缀)。类似于方法调用的位置参数和关键字参数
如上,config时位置参数不加–,work-dir是可选参数加–。调用脚本的时候可以:
python train.py configs/coop.py --work-dir ./logs
注意两个.py的区别,train.py是执行的脚本,coop.py是给其指定的配置文件,相当于train.py的argparse里config=coop.py