Hi, torch.utils.data.dataset.random_split returns a Subset object which has no transforms attribute. How can I split a Dataset object and return another Dataset object with the same transforms attribute? Thanks

Here is a small example: class MyDataset(Dataset): def __init__(self, subset, transform=None): self.subset = subset self.transform = transform def __getitem__(self, index): x, y = self.subset[index] if self.transform: x = self.transform(x…

Torch.utils.data.dataset.random_split

berat December 16, 2018, 8:10pm 3

How can I pass Subset to Dataset?

Issues with torch.utils.data.random_split

Why I am getting "AttributeError: 'Tensor' object has no attribute 'train_img' "

Individual transforms on subsets