时间戳

首页 > 解决方案 > RuntimeError: 4 维权重 96 3 11 11 的预期 4 维输入，但得到了大小为 [30, 50176] 的 2 维输入

python - RuntimeError: 4 维权重 96 3 11 11 的预期 4 维输入，但得到了大小为 [30, 50176] 的 2 维输入

问题描述

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

我在训练模型时遇到了错误-

RuntimeError: 4 维权重 96 3 11 11 的预期 4 维输入，但得到了大小为 [30, 50176] 的 2 维输入

这是 Colab 文件 -
任何人都可以建议我做错了什么。

标签： pythonpytorch

解决方案

推荐阅读