深度学习·Argparse

Argparse

命令行选项、参数和子命令解析器

ArgumentParser

命令行传参数->解析参数->获得对应参数

• 初始化：parser = argparse.ArgumentParser(description='xxx')
• 添加命令行参数： parser.add_argument("--training_filepath", type=str, help="Filepath to the training features", default="./data/2017/training/")
• 位置参数和可选参数：位置参数"training_filepath"必须添加；可选参数--training_filepath可以不添加，可以设定默认值
• 解析参数：args = parser.parse_args()，解析完参数后可以使用args获取并索引命令行参数
• 属性值：利用args.training_filepath获得对应属性`

NameSpace

from argparse import Namespace
就是一个结构体，可以储存一些参数或者变量的值，方便用.attribute获得

config=Namespace(
    project_name='DNN_Classifier',
    batch_size=32,
    learning_rate=1e-3,
    device=torch.device('cuda' if torch.cuda.is_available() else 'cpu'),
    epochs=50,
)

在dataloader=DataLoader(dataset,batch_size=config.batch_size,shuffle=True)中，利用config.batch_size获取变量值