训练准确率很高,预测准确率确很低的问题bug

使用VGG网络做信号灯树叶遮挡的图像分类问题:

最近接到一个二分类的任务,使用VGG训练的模型,结果发现训练准确率很高测试准确率去很低。不管怎么改权重改模型都一样。



原因分析:

经过长时间的分析确定了训练的代码没问题,权重也没问题,那么结果就出在了预测代码上。最终发现是权重没有加载,等于我每次都用初始化的模型来预测的。


解决方案:

大家遇到这种问题不妨看看权重有没有加载上

net.load_state_dict(model)
训练准确率很高,预测准确率确很低的问题bug_第1张图片

你可能感兴趣的:(pytorch,bug,机器学习,深度学习)