Загрузите наборы данных с torchvision. Различия между аргументами? - PullRequest
0 голосов
/ 18 июня 2020

Например, это написано для набора данных CelebA:

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

В то время как для CIFAR написано:

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

Мой вопрос: в чем разница между train=true и split=train?

1 Ответ

0 голосов
/ 18 июня 2020

Если вы ссылаетесь на ссылку , это зависит от используемого вами набора данных.

Например, для набора данных CIFAR у вас нет параметров split в документации, тогда как у CELEBA нет train параметров.

Это выбор, сделанный авторами библиотеки и, как вы можете прочитать в документации, в отношении этих параметров нет никаких норм, это действительно зависит от набора данных, который вы хотите использовать.

...