expected input to have 3 channels, but got 1 channels instead错误问题
从网上找的现成的代码,跑cifar10数据集没有问题,但是跑mnist数据集反而出了问题,是因为图片一个是彩色一个是灰度,在深度学习的时候,输入的channel不一样,找了很多资料,都是说让改成transforms.Lambda(lambda x: x.repeat(1,1,1)),还是不行,后来请教了同学,改成了transforms.Lambda(lambda x: x.repeat(3,1,1)),就OK了
...
原创
2022-04-05 00:23:20 ·
4891 阅读 ·
0 评论