Datawhale&kesci&伯禹教育-深度学习-第一次打卡3 训练集&读取数据

pytroch 自带的torchvision模块包

组织构成

1. torchvision.datasets: 一些加载数据的函数及常用的数据集接口；
2. torchvision.models: 包含常用的模型结构（含预训练模型），例如AlexNet、VGG、ResNet等；
3. torchvision.transforms: 常用的图片变换，例如裁剪、旋转等；
4. torchvision.utils: 其他的一些有用的方法。

get dataset

import torch
import torchvision 
import torchvision.transforms as transforms

mnist_train = torchvision.datasets.FashionMNIST(
root='/home/kesci/input/FashionMNIST2065', 
train=True, 
download=True, 
transform=transforms.ToTensor())
mnist_test = torchvision.datasets.FashionMNIST
(root='/home/kesci/input/FashionMNIST2065', 
train=False, 
download=True, 
transform=transforms.ToTensor())