如何用PyTorch代码示例演示MNIST手写数字分类？

本文共计762个文字，预计阅读时间需要4分钟。

torchvision包包含流行的数据集、模型结构和图像转换工具。torchvision.datasets包含以下数据集：MNIST、COCO（用于图像标注和目标检测）、LSUN（分类）。

torchvision包 包含了目前流行的数据集，模型结构和常用的图片转换工具。

torchvision.datasets中包含了以下数据集

MNIST
COCO（用于图像标注和目标检测）(Captioning and Detection)
LSUN Classification
ImageFolder
Imagenet-12
CIFAR10 and CIFAR100
STL10

torchvision.models

torchvision.models模块的 子模块中包含以下模型结构。