首页 分享 torch.backends.cudnn.benchmark和cudnn.deterministic

torch.backends.cudnn.benchmark和cudnn.deterministic

来源:花匠小妙招 时间:2024-12-16 13:50

torch.backends.cudnn.benchmark和cudnn.deterministic

最新推荐文章于 2024-10-25 20:59:09 发布

windistance 于 2020-10-29 16:58:37 发布

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

当使用新的尺度参数的时候,cuDNN 自动从几种算法里面寻找最适合当前配置的高效算法,之后所有相同参数的数据都采用这个算法。但是由于噪声等造成即使在同一个机器也可能会选择不同的算法。

因此方便复现、提升训练速度就:

torch.backends.cudnn.benchmark = False 1

不需要复现结果、想尽可能提升网络性能:

torch.backends.cudnn.benchmark = True 1

虽然通过torch.backends.cudnn.benchmark = False限制了算法的选择这种不确定性,但是由于,算法本身也具有不确定性,因此可以通过设置:
torch.backends.cudnn.deterministic = True

保持可复现性:

import numpy as np np.random.seed(0) torch.manual_seed(0) torch.cuda.manual_seed_all(0) torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False 1234567

https://pytorch.org/docs/stable/notes/randomness.html

相关知识

CV项目课程笔记02
和花和叶
大脑和神经系统和骨头...
菊花的花语和寓意和象征(菊花的花语和寓意和象征意义)
玫瑰花花语和寓意(玫瑰花花语和寓意和象征)
正则表达式中?=和?:和?!的理解
插花花名和含义和花语
康乃馨花语和寓意和象征
和花熊猫介绍 和花体形特征 和花家庭成员
一个等号(=)和三个等号和两个等号的区别(“===”和“==”)

网址: torch.backends.cudnn.benchmark和cudnn.deterministic https://www.huajiangbk.com/newsview1125386.html

所属分类:花卉
上一篇: 凌霄花如何种植?
下一篇: 花生种子的选购有哪些技巧? 爱问

推荐分享