RuntimeError: Given groups=1, weight of size 16 1 3 3, expected input[2, 3, 512, 512] to have 1 chan

pytorch训练代码出现这个错误,这个意思就是图像通道不对,要求1通道,给了3通道,去代码里改下就好了,或者改图像通道。

你可能感兴趣的:(pytorch)