小编Aak*_*W.S的帖子

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

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

这是我的代码

dataloader = torch.utils.data.DataLoader(dataset=dataset, batch_size=64)
images, labels = next(iter(dataloader))
Run Code Online (Sandbox Code Playgroud)

python pytorch dataloader

5
推荐指数
3
解决办法
1万
查看次数

标签 统计

dataloader ×1

python ×1

pytorch ×1