pytorch中argparse的用法

最新推荐文章于 2023-04-04 10:29:05 发布

浪子私房菜

最新推荐文章于 2023-04-04 10:29:05 发布

阅读量483

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/niuxuerui11/article/details/116203925

版权

pytorch 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

本文详细介绍了PyTorch中argparse模块的使用方法，通过实例展示了如何定义和解析命令行参数。重点讲解了add_argument方法，包括参数名、帮助信息、类型及默认值的设置。此外，还提到了在解析后如何直接添加新的参数值。对于需要处理命令行输入的PyTorch项目，argparse是一个不可或缺的工具。

摘要由CSDN通过智能技术生成

pytorch-argparse用法

import argparse

def parse():
		parser = argparse.ArgumentParser()
		parser.add_augment('---scales', help='scales', type='int', default=5)
		opt = parser.parser_args()
		return opt

if __name__ == '__main__':
		opt = parse()
		print (opt)
		opt.num = 'now two args'
		print (opt)

结果：

Namespace(scales=5)
Namespace(num=‘now two args’, scales=5)

note：使用argparse加载参数，在parse（）方法中定义完opt后，其他地方再用到其他参数，可以直接加参数，例如