Pytorch 从torchvision下载数据集。争论之间的分歧?
例如,它是为Pytorch 从torchvision下载数据集。争论之间的分歧?,pytorch,Pytorch,例如,它是为CelebA数据集编写的: torchvision.datasets.CelebA(root, split='train', download=False) 对于CIFAR,有以下内容: torchvision.datasets.CIFAR10(root, train=True, download=False) 我的问题是:train=true和split=train之间有什么区别?如果您引用它,则取决于您使用的数据集 例如,对于CIFAR数据集,文档中没有参数split,而CE
CelebA
数据集编写的:
torchvision.datasets.CelebA(root, split='train', download=False)
对于CIFAR,有以下内容:
torchvision.datasets.CIFAR10(root, train=True, download=False)
我的问题是:train=true
和split=train
之间有什么区别?如果您引用它,则取决于您使用的数据集
例如,对于CIFAR数据集,文档中没有参数split
,而CELEBA没有train
参数
这是编写库的人做出的选择,正如您在文档中看到的,这些参数没有任何规范,这实际上取决于您想要使用的数据集