python 保存argparse中的配置参数

保存argparse配置参数重要的一步是将parser.parse_args()转换为字典,再输出key,value进行保存。示例如下:

import argparse

parser = argparse.ArgumentParser(description='test')
parser.add_argument('--gpu_id', default=1, type=int)
parser.add_argument('--size', default=32, type=int)
parser.add_argument('--root', default='/users/xxx', type=str)
args = parser.parse_args()
print(args)
argsDict = args.__dict__
with open('setting.txt', 'w') as f:
    f.writelines('------------------ start ------------------' + '\n')
    for eachArg, value in argsDict.items():
        f.writelines(eachArg + ' : ' + str(value) + '\n')
    f.writelines('------------------- end -------------------')

如果对你有帮助,点个赞哦~

©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页