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

Question

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

这是我的代码

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

原文由 Aakanksha W.S 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以设置 batch_size=dataset.__len__() 以防数据集是 torch Dataset ，否则类似 batch_size=len(dataset) 的东西应该可以工作。

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

原文由 asymptote 发布，翻译遵循 CC BY-SA 4.0 许可协议