查看Dataset与DataLoader

class MyDataset(Dataset):
    def __init__(self):
    def __len__(self):
    def __getitem__(self, idx)：
        return sample

自定义的Dataset，因为实现了__getitem_ _，因此可以通过像访问列表似的访问Dataset的对象。

trainloader = torch.utils.data.DataLoader(dataset, batch_size=2, shuffle=True, num_workers=4)

loader可以通过for循环访问，得到的是batch_size个样本图片：torch.Size([2, 3, 448, 448])，如果Dataset中返回的是字典，那么就需要先将样本取出来。