直播小程式原始碼，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打亂的順序是相同，更多內容歡迎關注之後的文章