Python 中 parser.add_argument('--epochs') 的作用：设置训练轮数

parser.add_argument('--epochs', type=int, default=100, help='Number of epochs to train.') 的作用

该代码片段使用 Python 的 argparse 模块定义了一个名为 --epochs 的命令行参数，用于指定训练的轮数（即训练的迭代次数）。

• type=int: 指定参数类型为整数。
• default=100: 设置默认值为 100。
• help='Number of epochs to train.': 提供参数的帮助信息。

通过在命令行中使用 --epochs 参数并指定一个整数，例如 --epochs 200，您就可以覆盖默认值，将训练轮数设置为 200。

示例:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--epochs', type=int, default=100,
                        help='Number of epochs to train.')

args = parser.parse_args()

print(f'训练轮数: {args.epochs}')

输出:

如果在命令行中运行 python your_script.py --epochs 200，则输出将为：

训练轮数: 200