NN 模型每次都预测相同的事情

问题描述 投票:0回答:1

我正在使用 MNIST 数据集使用 pytorch 制作数字分类模型,当我训练和评估我的模型时,我获得了 97-99% 的准确率,但是当我保存和加载模型并通过模型传递某些内容时,预测输出是每次都是同样的事情。在本例中为 5。

我尝试用谷歌搜索它聊天gpt,它重新启动,但没有任何效果。我以为如果我重新开始,我就能解决问题,但不幸的是我遇到了同样的问题。

python pytorch neural-network mnist
1个回答
0
投票

我假设您正确保存和加载模型 对输入图像使用与训练中类似的转换

输出单个值通常是由于未转换输入数据造成的

© www.soinside.com 2019 - 2024. All rights reserved.