我正在使用 MNIST 数据集使用 pytorch 制作数字分类模型,当我训练和评估我的模型时,我获得了 97-99% 的准确率,但是当我保存和加载模型并通过模型传递某些内容时,预测输出是每次都是同样的事情。在本例中为 5。
我尝试用谷歌搜索它聊天gpt,它重新启动,但没有任何效果。我以为如果我重新开始,我就能解决问题,但不幸的是我遇到了同样的问题。
我假设您正确保存和加载模型 对输入图像使用与训练中类似的转换
输出单个值通常是由于未转换输入数据造成的