PyTorch卷积神经网络案例分析——LeNet

#LeNet 是整个卷积神经网络的开山之作,一共有七层,其实2层卷积和2层池化层交替出现, #最后输出三层全连接层得到整体的结果 import numpy as np import torch from torch.autograd import Variable from torch import nn, optim import matplotlib.pyplot as plt class LeNet(nn.Module): def __init__(self): super(LeNet, self).__init__() layer1 = nn.S