简单记一下 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) |