修改训练数据集:网络原本使用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

  1. 原因:通道数不匹配,MINIST使用灰度图像,CIFAR10使用RGB图像。
  2. 解决方式:加载图像时,做一下转换。
posted @ 2022-12-02 13:17  Kurie  阅读(2544)  评论(0编辑  收藏  举报