Предполагая pytorch 1.x+
, конструктор torchvision.datasets.MNIST следует этой подписи:
torchvision.datasets.MNIST(root, train=True, transform=None, target_transform=None, download=False)
Самый простой способ получить набор данных - установить download=True
, таким образом, он будет автоматически загружаться и магазин training.pt
и test.pt
. Предполагая локальную установку, по умолчанию они будут храниться где-то вроде .local/lib/python3.6/site-packages/torchvision/
, хотя вам не нужно об этом беспокоиться.