修复火炬种子random_split（）

Question

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

Answer 1

您可以使用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中看到）。