从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

这是您想要的吗？希望如此：）