Python 运行时错误：四维权重96 3 11 11的预期四维输入，但得到的是大小为[30，50176]的二维输入_Python_Pytorch

Python 运行时错误：四维权重96 3 11 11的预期四维输入，但得到的是大小为[30，50176]的二维输入

python pytorch

Python 运行时错误：四维权重96 3 11 11的预期四维输入，但得到的是大小为[30，50176]的二维输入,python,pytorch,Python,Pytorch,我对我正在处理的一个问题有疑问。我正试图在google colab上的At&t ORL人脸数据集上使用Alexnet 我在训练模型时出错了- 运行时错误：四维权重96 3 11 11的预期四维输入，但得到的是大小为[30，50176]的二维输入这是Colab文件- 谁能告诉我我做错了什么。这能回答你的问题吗？您需要添加一个单例“批处理”维度，这是否回答了您的问题？您需要添加一个单例“批处理”维度

我对我正在处理的一个问题有疑问。我正试图在google colab上的At&t ORL人脸数据集上使用Alexnet

我在训练模型时出错了-

运行时错误：四维权重96 3 11 11的预期四维输入，但得到的是大小为[30，50176]的二维输入

这是Colab文件-

谁能告诉我我做错了什么。

这能回答你的问题吗？您需要添加一个单例“批处理”维度，这是否回答了您的问题？您需要添加一个单例“批处理”维度