文章目录
MNIST数据集是学习各种神经网络的开胃菜。它的任务是识别手写数字(1~9),具体介绍官网http://yann.lecun.com/exdb/mnist/。这次我们用MNIST来小试牛刀。上次我们采用的是MLP网络,这次使用CNN网络。
1. 准备工作
1.1 导包和设置随机种子
import torchvision
import torchvision.datasets as datasets #为了下一步加载数据
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd