使用pytorch训练好的模型对两类照片进行预测,预测结果均为后一种类别

看看是不是样本不平衡，或者你的损失函数、激活函数问题，或者是训练不完全。

训练的时候，分类的准确率高吗?

【相关推荐】

这个问题的回答你可以参考下: https://ask.csdn.net/questions/7480970
你也可以参考下这篇文章：使用pytorch训练好的模型来预测单张照片类别错误率几乎百分百的原因和解决方法
同时，你还可以查看手册：pytorch 使用PyTorch进行培训中的内容
您还可以看一下龙良曲老师的深度学习与PyTorch入门实战教程课程中的常见函数的梯度小节, 巩固相关知识点
除此之外, 这篇博客: pytorch 训练正常,测试显存爆掉问题解决中的 pytorch训练正常但是在预测的时候显存爆掉,错误信息如下 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
```
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
```

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^