在使用parser过程中经常遇到
parser.parse_args 出现systemexit 2
具体如下:
parser = argparse.ArgumentParser()
# model arguments
parser.add_argument('--img_size', type=int, default=256,
help='Image resolution')
args = parser.parse_args()
vscode调试时必然会出现这个问题,但是利用命令 python mian.py --img_size 256 运行就没问题
然后发现,改成函数的形式在调试过程中就不会出问题了
修改如下:
def get_args():
parser = argparse.ArgumentParser()
# model arguments
parser.add_argument('--img_size', type=int, default=256,
help='Image resolution')
return parser.parse_args()
args = get_args()
具体的原理不清楚,但就是可以用
我试了网上的另一种方法,
args = parser.parse_args(args=[])
不管用。。。