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

新手上路,请多包涵

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

这是我的代码

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

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

阅读 452
1 个回答

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

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

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

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题