是否改组了torchvision.datasets.MNIST和torchvision.datasets.FashionMNIST？

Question

Torchvision中的MNIST和FashionMNIST数据集是否已经改组？即，如果我这样做的话

from torchvision.datasets import MNIST
ds = MNIST(".", download=True, train=True)
train_samples = [ds[i] for i in range(50000)]
val_samples = [ds[i] for i in range(50000, 60000)]

将此拆分为train和val可以，因为数据集已经被改组了，还是不好？那么FashionMNIST呢？

Answer 1

您需要了解数据加载器。数据加载器为您执行改组，批处理等操作。

Here是一个教程。

[如果您使用的是TorchVision的数据集（而不是编写自己的数据集类），则很可能已经有一个train作为布尔参数，可用于指定是训练组还是测试设置。