Cuando se ejecuta un programa de Python en el servidor, a menudo es necesario ingresar parámetros adicionales, o esperar que estos parámetros no necesiten ser cambiados directamente en el código. Es una buena manera de usar el módulo incorporado argparse.
Pasos de uso
Solo necesita 3 pasos para completar:
1. Crea un objeto
parser = argparse.ArgumentParser(description='example')
2. Incrementar los parámetros
parser.add_argument('--no-cuda', action='store_true', default=False,
help='disables CUDA training')
3, análisis
args = parser.parse_args()
print(args.cuda)
Correspondiente a argarse, es necesario aclarar el significado de los parámetros en el paso 2. El que comienza con - indica parámetros opcionales, si no lo tiene debe agregarlo, de lo contrario se reportará un error.
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)
Algunos usos comunes:
Compruebe si hay una GPU, si la hay, enciéndala:
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!')
Establezca los parámetros predeterminados de los parámetros:
parser.add_argument ('- var_bag_length', type = int, default = 2, metavar = 'VL',
help = 'varianza de la longitud de la bolsa')
--var_bag_length: indica que el nombre de la variable es var_bag_length, que es un parámetro opcional
type = int: expresado como tipo int
default = 2: significa que var_bag_length es 2 por defecto
metavar = 'VL': indica que el uso promueve la variable de usuario a VL
help = 'varianza de la longitud de la bolsa': información de ayuda