在服务器上运行python程序时候,往往需要输入额外的参数,或者希望这些参数不用直接在代码里面改变,使用argparse内置模块不失为一种好办法。
使用步骤
使用仅仅需要3个步骤就可以完成:
1,创建对象
parser = argparse.ArgumentParser(description='example')
2,增加参数
parser.add_argument('--no-cuda', action='store_true', default=False,
help='disables CUDA training')
3 ,解析
args = parser.parse_args()
print(args.cuda)
对应argarse,需要搞清楚第2步里面参数意义,带--开头,表示可选参数,不带则必须加入,否则会报错,
parser = argparse.ArgumentParser()
parser.add_argument("add", help="display a square of a given number", type=int)
args = parser.parse_args()
print(args.add+2)
一些常用用法:
检测是否有GPU,如果有则打开:
parser = argparse.ArgumentParser(description='example')
parser.add_argument('--no-cuda', action='store_true', default=False,
help='disables CUDA training')
args = parser.parse_args()
args.cuda = not args.no_cuda and torch.cuda.is_available()
if args.cuda:
print('\nGPU is ON!')
设置参数默认参数:
parser.add_argument('--var_bag_length', type=int, default=2, metavar='VL',
help='variance of bag length')
--var_bag_length :表示变量名字为var_bag_length,是可选参数
type=int :表示为int类型
default=2 :表示 var_bag_length默认为2
metavar='VL' :表示usage提升用户变量为VL
help='variance of bag length' :帮助信息