修改训练数据集:网络原本使用MNIST数据集进行训练,现改为使用CIFAR10来进行训练。报错:RuntimeError: Given groups=1, weight of size [6, 1, 13, 13], expected input[1, 3, 32, 32] to have 1 channels, but got 3 channels instead
- 原因:通道数不匹配,MINIST使用灰度图像,CIFAR10使用RGB图像。
- 解决方式:加载图像时,做一下转换。
curie.