torch.utils.data.DataLoader同时遍历两个dataload

torch.utils.data.DataLoader同时遍历两个dataloader

在MNIST数据集中，我想给train_dataset加上噪声。用enumerate遍历的时候，如果需要两个dataloader一起遍历，需要加上zip(dataloader1,dataloader2)。train_loader1,Rd_loader1都是600✖100，如下：

for i, ((images, labels),rd) in enumerate(zip(train_loader1,Rd_loader1)):

这是两者维数相同的情况，如果不同，则用cycle(dataloader1,dataloader2)。train_loader1是600✖100，Rd_loader1是200✖100，如下：

for i, ((images, labels),rd) in enumerate(cycle(train_loader1,Rd_loader1)):

防止自己忘记，记录一下。