python - 如何从 PyTorch 中的数据加载器获取整个数据集

Question

如何从 DataLoader 加载整个数据集？我只得到一批数据集。

这是我的代码

dataloader = torch.utils.data.DataLoader(dataset=dataset, batch_size=64)
images, labels = next(iter(dataloader))

标签： pythonpytorchdataloader

Answer 1

batch_size=dataset.__len__()如果 dataset 是 torch ，您可以设置Dataset，否则batch_szie=len(dataset)应该可以使用。

请注意，这可能需要大量内存，具体取决于您的数据集。