Pytorch Dataset & Dataloader

class MyDataset(Dataset):
    # read data & preprocess
    def __init__(self, file):
        self.data = ...
    
    # returns one sample at a time
    def __getitem__(self, index):
        return self.data[index]
    
    # returns the size of dataset
    def __len__(self):
        return len(self.data)
dataset = MyDataset(file)
dataloader = Dataloader(dataset, batch_size, shuffle = True)

Dataloader中的shuffle在Training 中为 True，在Testing 中为False
未完待续 。 。 。