例子tes.py
import argparse
# 实例化命令行解析器对象
parser = argparse.ArgumentParser(description='Demo of argparse')
# 添加变量
parser.add_argument('--epochs', type=int, default=30)
parser.add_argument('--batch', type=int, default=4)
# 解析添加的变量
args = parser.parse_args()
print(args.epochs, args.batch)
运行结果:
30 4
命令行运行:
python tes.py --epoch 40 --batch 40
运行结果:
40 40