Hello, I am trying to create a dataloader with the following code:
data_loader = DataLoader(dataset, batch_size=1, num_workers=num_workers, sampler=SubsetRandomSampler(pool_idx))
where pool_idx =
pool_idx = random.sample(range(1, len(test_dataset)), pool_size)
Example of pool_idx:
[9, 6, 2, 1, 3, 5, 7, 4, 8, 0]
Is it possible to have the image + each corresponding id so when iterating through the dataset I can have access to each id?