python 参数解析器argparse

#1导入库
import argparse
#2创建解析器
parser = argparse.ArgumentParser()
#3添加参数
parser.add_argument('--num_classes', type=int, default=4)
parser.add_argument('--epochs', type=int, default=10)
parser.add_argument('--batch-size', type=int, default=8)
parser.add_argument('--lr', type=float, default=5e-4)
parser.add_argument('--wd', type=float, default=5e-2)
#4解析参数
opt = parser.parse_args()
#5调用参数
main(opt)