MNIST数据集是学习各种神经网络的开胃菜。它的任务是识别手写数字(1~9),具体介绍官网http://yann.lecun.com/exdb/mnist/。这次我们用MNIST来小试牛刀。
1. 准备工作
1.1 导包和设置随机种子
import torchvision
import torchvision.datasets as datasets #为了下一步加载数据
import torch
import torch.nn.functional as F
import numpy as np
import random #设置随机数
USE_CUDA = torch.cuda.is_available() #GPU可用的标志位
random.seed(1)
np.random.seed(1)
torch.manual_seed(1)
if USE_CUDA:
torch.cuda.manual_seed(1)
1.2 加载数据和归一化
train_mnist = datasets.MNIST(root = './data', train = True, download = True) #训练集
test_mnist = datasets.MNIST(root = './data', train = False, d