直播小程序源码,pytorch同时让两个dataloader打乱的顺序是相同
直播小程序源码,pytorch同时让两个dataloader打乱的顺序是相同
class MyDataset(Dataset):
def __init__(self, datasetA, datasetB):
self.datasetA = datasetA
self.datasetB = datasetB
def __getitem__(self, index):
xA = self.datasetA[index]
xB = self.datasetB[index]
return xA, xB
def __len__(self):
return len(self.datasetA)
datasetA = ...
datasetB = ...
dataset = MyDataset(datasetA, datasetB)
loader = DataLoader(dataset, batch_size=10, shuffle=True)
以上就是 直播小程序源码,pytorch同时让两个dataloader打乱的顺序是相同,更多内容欢迎关注之后的文章