argparse模块介绍

    在服务器上运行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' :帮助信息

add_argument() 方法参数可以看这篇

猜你喜欢

转载自blog.csdn.net/qq_39463175/article/details/108083392