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