在使用 dataloader 测试数据集时，我们应该设置 shuffle=true 还是没有关系？

Question

我有一个自定义数据集（披萨、寿司和牛排的图像）。 我正在使用 torch DataLoader ，现在在编写测试数据加载器自定义时，我们应该设置 shuffle=true 还是没关系？？

我还没看出区别，只是问问将军。

Answer 1

如果您使用

shuffle=True

，它应该在每个时期重新整理数据（[来自 PyTorch 文档][1]）。这意味着在每个时期，数据批次的排列方式将与前一个批次不同。

在测试过程中，由于在测试过程中模型不会改变，并且您通常不会测试多个纪元，因此应该不会有任何区别。

但是，设置

shuffle=False

可以确保预测顺序的一致性，并且可以使调试更容易。 [1]：https://pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader