简单记一下 argparse 的使用,老是忘。

下面是一个使用 argparse 的小例子:

# 基于 ChatGPT 生成
import argparse
# 创建 ArgumentParser 对象
parser = argparse.ArgumentParser(description='Example script with command line arguments.')
# 添加命令行参数
parser.add_argument('--lr', type=float, default=0.001, help='learning rate')
parser.add_argument('--epochs', type=int, default=10, help='number of training epochs')
parser.add_argument('--batch_size', type=int, default=32, help='batch size')
parser.add_argument('--abc', type=int, default=1, help='an example parameter')
# 解析命令行参数
args = parser.parse_args()
# 访问参数的值
print("Learning Rate:", args.lr)
print("Epochs:", args.epochs)
print("Batch Size:", args.batch_size)
print("ABC Parameter:", args.abc)