Pytorch入门实战-----CNN识别手写数据集

CNN网络对图像的提取特征确实厉害了，只跑一个epoch识别率就达到了98。

import torch
import torch.nn as nn
import torchvision.transforms as transforms
import torchvision.datasets as dsets
from torch.autograd import Variable

batch_size = 100
learning_rate = 0.01
num_epochs = 1

train_datasets = dsets.MNIST(root='./data',
                             train=True,
                             transform=transforms.ToTensor(),
                             download=False)
test_datasets = dsets.MNIST(root='./data',
                            train=False,
                            transform=transforms.ToTensor(),
                            download=False)
train_loader = torch.utils.data.DataLoader(dataset=train_datasets,