修复火炬 random_split() 的种子

Question

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

Answer 1

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

import torch
torch.manual_seed(0)

有关更多信息，请参阅重现性文档。

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

torch.manual_seed(torch.initial_seed())

AFAIKpytorch 不提供像seed或random_state这样的参数（例如可以看到sklearn）。

Answer 2

正如您从文档中看到的，可以将生成器传递给 random_split

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