argparse:
import argparse
parser = argparse.ArgumentParser(description='Hello World!')
parser.add_argument('--gpus', type=str, default = None)
parser.add_argument('--batch_size', type=int, default=64)
parser.add_argument('--isRandom', type=bool, default=64)
args = parser.parse_args()
print args.gpus
print args.batch_size
命令行使用如下:
python test.py --gpu=1 --batch_size=64 --isRandom=False
需要注意的是,脚本运行命令python test.py时,
--isRandom False
会被自动解析成True。因为argparse模块传参时,只有传空值给布尔值的参数才会是True值,其他字符串传的都是False值。