Hello. This is my CustomDataSetClass: class CustomDataSet(Dataset): def __init__(self, main_dir, transform): self.main_dir = main_dir self.transform = transform all_imgs = os.listdir(main_dir) self.total_imgs = natsort.natsorted(all_imgs) for file_name in…

Yes, the order should be preserved as shown in this simple example using TensorDatasets: datasets = [] for i in range(3): datasets.append(TensorDataset(torch.arange(i*10, (i+1)*10))) dataset = ConcatDataset(datasets) loader = DataLoader( dataset, shuffle=False, num_workers=0, b…

PyTorch Forums

How does ConcatDataset work?

ptrblck May 17, 2022, 4:30am 10

Double post from here.

show post in topic

Home
Categories
FAQ/Guidelines
Terms of Service
Privacy Policy

Powered by Discourse, best viewed with JavaScript enabled