2_dataset, dataloader

dataset, dataloader

torchvision.datasets里面集成了一些常见的数据集，例如MNIST和CIFAR10

1) Dataset

以MNIST为例，其使用方式如下

import torch
import torchvision
from torchvision import transforms

train_dataset = torchvision.datasets.MNIST(root='../data',
                                           train=True
1
2
3
4
5