Pytorch中的argparse模块用法

在 PyTorch 中，argparse 模块通常用于解析命令行参数，使得脚本更加灵活和可配置。通过 argparse，你可以为脚本添加参数，使得在运行时可以轻松地调整超参数、文件路径等配置项，而无需修改代码本身。

1. 安装和导入 argparse

argparse 是 Python 标准库的一部分，因此无需额外安装。你只需在脚本中导入它即可：

import argparse

2. 创建解析器

你需要创建一个 ArgumentParser 对象，它用于存储将要解析的所有信息：

parser = argparse.ArgumentParser(description="PyTorch Training Script")

3. 添加参数

使用 add_argument 方法为解析器添加参数。你可以指定参数名称、类型、默认值和帮助信息：

parser.add_argument('--batch-size', type=int, default=64, help='input batch size for training (default: 64)')
parser.add_argument('--epochs', type=int, default=10, help='number of epochs to train (default: 10)')
parser.add_argument('--learning-rate'