修复torch random_split() 的种子

Question

是否可以在拆分数据集时修复 torch.utils.data.random_split() 的种子，以便重现测试结果？

Answer 1

从documentation 中可以看出，可以将生成器传递给 random_split

random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))

Answer 2

您可以使用torch.manual_seed 函数全局播种脚本：

import torch
torch.manual_seed(0)

更多信息请参见reproducibility documentation。

如果您想专门播种torch.utils.data.random_split，您可以在之后将种子“重置”为其初始值。只需像这样使用torch.initial_seed()：

torch.manual_seed(torch.initial_seed())

AFAIK pytorch 不提供seed 或random_state 之类的参数（例如，可以在sklearn 中看到）。