修改训练数据集:网络原本使用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.
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· .NET10 - 预览版1新功能体验(一)