Pytorch的Reproducibility(可复现性)

Posted yanxingang

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Pytorch的Reproducibility(可复现性)相关的知识,希望对你有一定的参考价值。

对于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(可复现性)的主要内容,如果未能解决你的问题,请参考以下文章

pytorch-pretrained-BERT:BERT PyTorch实现,可加载Google BERT预训练模型

PyTorch 中是不是存在干净且可扩展的 LSTM 实现? [关闭]

如何通过 pytorch 使 Intel GPU 可用于处理?

具有多个工作人员的可迭代 pytorch 数据集

如何在 Keras 模型中实现一些可训练的参数,例如 Pytorch 中的 nn.Parameters()?

可重复使用的Pytorch结果和随机种子