Torch使两个Dataloader保持一致的打乱顺序

背景

PYG中，数据类无法自定义，有时候要汇聚多组图数据，此时要求Dateloader的打乱顺序一致使得他们输出相同。

方法

使用相同的随机数种子对数据进行打乱。

代码

from torch_geometric.loader import DataLoader
import random
seed=random.randint(0,99999999)
print(seed)
torch.manual_seed(seed)
train_data=train_data.shuffle()
torch.manual_seed(seed)
seq_train_data=seq_train_data.shuffle()
# 创建dataloader1对象
dataloader1 = DataLoader(train_data, batch_size=32, shuffle=False)
# 使用相同的随机种子创建dataloader2对象
dataloader2 = DataLoader(seq_train_data,batch_size=32, shuffle=False)
for data in zip(dataloader1,dataloader2):
    print(data[0].y)
    print(data[1].y)
    break