例:CycleGAN and pix2pix in PyTorch - Github
看到 train.py 开始有一行传入参数的代码:opt = TrainOptions().parse()
查看 TrainOptions
及其基类 BaseOptions
的定义,将所有带有 required=True
的参数都修改为包含默认配置的参数,这样就可以不用传参直接运行项目比如:
原始:
parser.add_argument('--dataroot', required=True, help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')
修改为:
parser.add_argument('--dataroot', default="./dataset", help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')