2_dataset, dataloader

dataset, dataloader

torchvision.datasets里面集成了一些常见的数据集,例如MNISTCIFAR10

1) Dataset

MNIST为例,其使用方式如下

import torch
import torchvision
from torchvision import transforms

train_dataset = torchvision.datasets.MNIST(root='../data',
                                           train=True

你可能感兴趣的:(Pytorch深度学习,深度学习,python,人工智能)