argparser参数配置模块简介

argparse模块是Python内置的用于命令项选项与参数解析的模块。不仅可在python文件中将算法参数集中放到一起，并且可在命令行/终端进行指定参数修改，非常适合深度学习中超参数的设置与修改。

import argparse

---

1 快速入门

基本操作

1. 导入argparse包
2. 创建argparse.ArgumentParser对象parser，即参数解析器/解析对象
3. 用add_argument方法往parser中添加参数
4. 用parse_args方法解析参数，所得的参数对象可正常用.运算符调用各参数

示例

# quick_start.py
import argparse


parser = argparse.ArgumentParser()

parser.add_argument('--batch_size', type=int, default=64, help='num of batches')
parser.add_argument('--num_window', type=int, default=5, help='num of windows')
parser.add_argument('--weight', type=str, default=r'../pretrain.pth', help='path of the pretrained model')

args = parser.parse_args()
print(args.batch_size)

终端操作

python quick_start.py --batch_size 128

---

2 初始化参数解析器 ArgumentParser

参数解析器对象ArgumentParser保存了所有必要的信息，用以将命令行参数解析为相应的python数据类型。

# main.py
import argparse

# 创建参数解析器：一个用于