创建一个参数解释器:
import argparse
parser = argparse.ArgumentParser()
配置参数:
parser.add_argument('--max_epochs', type=int, default=200)
值得说明的是:
如果要输入List类型的参数,应该是
parser.add_argument('--decay_epochs', nargs='+', type=int, default=[90, 120])
并且在传入参数时应该是
--decay_epochs 120 200
具体可参见:
https://blog.csdn.net/kinggang2017/article/details/94036386
读取参数:
args = parser.parse_args()
a = args.max_epochs