python parser.add_argument

parser = argparse.ArgumentParser()

parser.add_argument('--path', '--checkpoints', type=str, default='./checkpoints', help='model checkpoints path (default: ./checkpoints)')

parser.add_argument('--model', type=int, choices=[1, 2, 3, 4], help='1: edge model, 2: inpaint model, 3: edge-inpaint model, 4: joint model')

args = parser.parse_args()

一、这里出现两个'--path', '--checkpoints',测试发现,命令行输入:(正确)

1.python train.py --model 2 --path ./checkpoints/places2

2.python train.py --model 2 --checkpoints ./checkpoints/places2

但是程序中,必须使用第一个参数,即args.path。使用args.checkpoints错误。

二、如果是一长一短:

parser.add_argument('--path', '-p', type=str, default='./checkpoints', help='model checkpoints path (default: ./checkpoints)')

命令行输入:(正确)

1.python train.py --model 2 --path ./checkpoints/places2

2.python train.py --model 2 -p ./checkpoints/places2

但是程序中只能使用,args.path

你可能感兴趣的:(ubuntu)