Custom dataset based on CIFAR10

[thecho7 (Suho Cho)]

It seems self.dst_train is not a tensor but a list.
From the example which is @ptrblck mentioned,

How to use BatchSampler with __getitem__ dataset

Maybe this code snippet might be helpful: class MyDataset(Dataset): def __init__(self): self.data = torch.arange(100).view(100, 1).float() def __getitem__(self, index): x = self.data[index] return x def __len__(self): return len(self.data) dataset = MyDataset() sampler = torch.utils.data.sampler.BatchSampler( torch.utils.data.sampler.RandomSampler(dataset), batch_size=10, drop_last=False) loader = DataLoader( da…

The dataset is a tensor has size of (100, 1) but your self.dst_train is a list.
List type cannot be indexed with list where a tensor can do.

Try convert self.dst_train to a tensor.
Should work