pytorch预测之解决多次预测结果不一致问题

更新时间：2021年06月01日 08:40:15 作者：confusingbird

这篇文章主要介绍了pytorch多次预测结果不一致的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

为什么多次预测结果不一致

1、检查是否在每次预测前使用

model.eval()

或者是

with torch.no_grad():
   for ...

推荐下面的方法，上面的的方法计算梯度，但是并不反向传播，下面的方法既不计算梯度，也不反向传播，速度更快。

2、检查是否取消了所有的dropout

3、设置随机种子

def setup_seed(seed):
    np.random.seed(seed)
    random.seed(seed)
    torch.manual_seed(seed) #cpu
    torch.cuda.manual_seed_all(seed)  #并行gpu
    torch.backends.cudnn.deterministic = True  #cpu/gpu结果一致
    torch.backends.cudnn.benchmark = True   #训练集变化不大时使训练加速

4、保证实例化模型前要将is_training置为false；这两行代码顺序不能颠倒

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Pytorch中TensorDataset与DataLoader的使用方式
这篇文章主要介绍了Pytorch中TensorDataset与DataLoader的使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-09-09
python gui开发——制作抖音无水印视频下载工具(附源码)
这篇文章主要介绍了python gui开发——制作抖音无水印视频下载工具(附源码)的相关资料，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下
2021-02-02
python实现简单成绩录入系统
这篇文章主要为大家详细介绍了python实现简单成绩录入系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-09-09
Python操作Sonqube API获取检测结果并打印过程解析
这篇文章主要介绍了Python操作Sonqube API获取检测结果并打印过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-11-11
Python使用xpath对解析内容进行数据提取
XPath 使用路径表达式来选取HTML/ XML 文档中的节点或节点集,节点是通过沿着路径 (path) 或者步 (steps) 来选取的,本文将给大家介绍Python使用xpath对解析内容进行数据提取的方法,需要的朋友可以参考下
2024-05-05
python创建只读属性对象的方法(ReadOnlyObject)
有时需要创建一个带只读属性的对象，大家可以参考下如下的方法进行创建，稍加改造，可以得到很特殊的效果
2013-02-02
python 如何将数据写入本地txt文本文件的实现方法
这篇文章主要介绍了python 如何将数据写入本地txt文本文件的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-09-09
理想高通滤波实现Python opencv示例
今天小编就为大家分享一篇关于理想高通滤波实现Python opencv示例，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
2019-01-01
Python类的多继承的搜索顺序
python和C++一样,支持多继承,本文主要介绍了Python类的多继承的搜索顺序,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧
2024-03-03
python爬虫刷访问量 2019 7月
这篇文章主要介绍了python爬虫刷访问量 2019 7月，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-08-08