zi2zi-pytorch
zi2zi-pytorch copied to clipboard
RuntimeError: output with shape [1, 256, 256] doesn't match the broadcast shape [3, 256, 256]
train.py阶段
我也碰到了,解决不了,头疼
把train.py第45行 : "parser.add_argument('--input_nc', type=int, default=3, help='number of input images channels')" default的值改成1就行了。 这个值是dataloader识别的图片的通道数,之前我们的数据集是单通道的,这里得改成1
input_nc的问题,几个地方默认值不一样
把train.py第45行 : "parser.add_argument('--input_nc', type=int, default=3, help='number of input images channels')" default的值改成1就行了。 这个值是dataloader识别的图片的通道数,之前我们的数据集是单通道的,这里得改成1
感恩大哥啊 救我老命