从torchvision下载数据集。争论之间有区别吗？

Question

例如，它是为CelebA数据集编写的：

torchvision.datasets.CelebA(root, split='train', download=False)

在编写CIFAR时，写着：

torchvision.datasets.CIFAR10(root, train=True, download=False)

我的问题是：train=true和split=train有什么区别？

Answer 1

如果您引用link，则取决于您使用的数据集。

例如，对于CIFAR数据集，文档中没有参数split，而CELEBA没有train参数。

这是由编写该库的人做出的选择，正如您可以在文档中阅读的那样，这些参数没有规范，这实际上取决于您要使用的数据集。