torch.utils.data.DataLoader可以设置随机数种子吗
时间: 2023-07-23 22:18:40 浏览: 147
Pytorch在dataloader类中设置shuffle的随机数种子方式
是的,torch.utils.data.DataLoader可以设置随机数种子。可以通过设置torch.manual_seed()函数来设置随机数种子,具体代码如下:
```
import torch
import torch.utils.data as Data
torch.manual_seed(1234) # 设置随机数种子
dataset = Data.TensorDataset(torch.Tensor([1,2,3]), torch.Tensor([4,5,6]))
dataloader = Data.DataLoader(dataset, batch_size=2, shuffle=True)
for batch_x, batch_y in dataloader:
print(batch_x, batch_y)
```
在上面的代码中,通过设置torch.manual_seed(1234)函数来设置随机数种子。这样,每次运行程序时,生成的随机数序列都是相同的,确保了程序的可重复性。
阅读全文