加入的隐藏层中100会报错，由于496*13*13*1才能匹配，这里应该是13吧

来源：4-4 利用神经网络解决分类和回归问题（2）

weixin_慕无忌0346932

2022-03-08

写回答

1回答

会写代码的好厨师

2022-03-16

这个问题应该说的线性层的参数定义？

线性层的输入数据需要将数据的维度reshape成 batchsize * n，这里的n就是对应的infeature的长度。

如果加100报错，可以检查下数据输入的维度是不是和视频中一致，如果有区别，则需要进行调整。如果是496*13*13*1的话，可以尝试改成13*13 试试。

PyTorch入门到进阶实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发，快速掌握PyTorch框架

1190 学习 · 298 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1