Распечатать случайную выборку из загрузчика данных в PyTorch

Question

У меня есть определенный набор данных, загруженный в загрузчик данных. Например, если я хочу сохранить 100 изображений из этого загрузчика данных, как мне перебрать загрузчик данных, чтобы сохранить их?

Answer 1

Я не совсем уверен, что вы пытаетесь сделать (возможно, отредактируйте свой вопрос), но, возможно, это поможет:

dataset = Dataset()
dataloader = torch.utils.data.DataLoader(
                 dataloader,
                 batch_size=32,
                 num_workers=1,
                 shuffle=True)

for samples, targets in dataloader:
    # 'sample' now is a batch of 32 (see batch-size above) elements of your dataset

Это то, что вы хотели? Надеюсь на это :)