ray.train.torch.enable_reproducibility

ray.train.torch.enable_reproducibility(seed: int = 0) → None

限制非确定性行为的来源。

这个函数：

• 种子 PyTorch、Python 和 NumPy。

• 禁用 CUDA 卷积基准测试。

• 配置 PyTorch 使用确定性算法。

• 为多进程数据加载生成的种子工作者。

参数:

seed – 用于种子库和数据工作者初始化的数字。

警告

train.torch.enable_reproducibility() 不能保证在不同执行中完全重现结果。要了解更多信息，请阅读 PyTorch 关于随机性的说明。