码迷,mamicode.com
首页 > 其他好文 > 详细

Pytorch的Reproducibility(可复现性)

时间:2019-04-02 10:55:28      阅读:237      评论:0      收藏:0      [点我收藏+]

标签:family   mini   相同   code   mil   uda   不能   结果   cos   

对于Pytorch的结果复现,在不同版本的pytorch或者不同的硬件平台之间,不能够保证结果是完全一致的,即使在CPU和GPU上采用相同的随机数种子。

关于torch.backend.cudnn.benchmark, (CuDNN backend是什么

torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

是相对来说确定性的模式。

包括Conv2d这样的函数,In some circumstances when using the CUDA backend with CuDNN, this operator may select a nondeterministic algorithm to increase performance. If this is undesirable, you can try to make the operation deterministic (potentially at a performance cost) by setting torch.backends.cudnn.deterministic = True.

Pytorch的Reproducibility(可复现性)

标签:family   mini   相同   code   mil   uda   不能   结果   cos   

原文地址:https://www.cnblogs.com/yanxingang/p/10640878.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!