test_loader = torch.utils.data.DataLoader(image, batch_size=1, shuffle=False, num_workers=0)如何避免乱序遍历
时间: 2023-07-15 21:14:40 浏览: 173
解决pytorch DataLoader num_workers出现的问题
5星 · 资源好评率100%
如果你想按顺序遍历数据集,只需要将 `shuffle` 参数设置为 `False` 即可。在你的例子中,将 `shuffle` 参数设置为 `False`,即可按顺序遍历数据集:
```
test_loader = torch.utils.data.DataLoader(image, batch_size=1, shuffle=False, num_workers=0)
```
另外,如果你想要并行加载数据,可以将 `num_workers` 参数设置为一个大于 0 的整数,这将使用多个进程来加载数据,加速数据加载的过程。但是需要注意的是,使用多个进程加载数据时,数据顺序可能会发生变化。
阅读全文