如何从 DataLoader 加载整个数据集?我只得到一批数据集。
这是我的代码
dataloader = torch.utils.data.DataLoader(dataset=dataset, batch_size=64)
images, labels = next(iter(dataloader))
原文由 Aakanksha W.S 发布,翻译遵循 CC BY-SA 4.0 许可协议
如何从 DataLoader 加载整个数据集?我只得到一批数据集。
这是我的代码
dataloader = torch.utils.data.DataLoader(dataset=dataset, batch_size=64)
images, labels = next(iter(dataloader))
原文由 Aakanksha W.S 发布,翻译遵循 CC BY-SA 4.0 许可协议
2 回答5.2k 阅读✓ 已解决
2 回答1.1k 阅读✓ 已解决
4 回答1.4k 阅读✓ 已解决
3 回答1.3k 阅读✓ 已解决
3 回答1.3k 阅读✓ 已解决
2 回答884 阅读✓ 已解决
1 回答1.8k 阅读✓ 已解决
您可以设置
batch_size=dataset.__len__()
以防数据集是 torchDataset
,否则类似batch_size=len(dataset)
的东西应该可以工作。请注意,这可能需要大量内存,具体取决于您的数据集。