首页 / 资讯中心 / 文章详情

R torchvision 0.8.0 中 MNIST 张量缺少通道维的修复

R torchvision 0.8.0 中 MNIST 张量缺少通道维的修复 ★ FEATURED ARTICLE
用 R 的torch+luz跑 MNIST 卷积神经网络时,训练一开始就崩在第一个卷积层,报错信息把 batch size 当成了通道数。这不是模型写错,而是数据加载器返回的张量形状变了。问题现象模型定义中卷积层的输入通道为 1:self$conv1=nn_conv2d(1,32,3)# N * 1 * 28 * 28 - N * 32 * 26 * 26调用fit()时报错:Error in (function (input, weight, bias, stride, padding, dilation, group
阅读完成 · 觉得有帮助?
咨询建站