上次课我们讲解了对于CIFAR10数据读取部分代码的编写,本节讲解如何编写经典的LeNet5神经网络。
首先创建python文件,命名LeNet5。
?...nn.Sequential(
# 把网络结构放在Sequential中十分方便
)
下面把网络结构放在Sequential中
注意因为CIFAR10...# 由于不清楚是什么维度变到了120,这里暂时先写2
nn.ReLU(),
nn.Linear(120, 64),
nn.ReLU(),
nn.Linear(84, 10...),
)
为能知道具体的维度信息,这里可以先构建一个随机的假数据,代入其中先行进行计算。...nn.ReLU(),
nn.Linear(120, 84),
nn.ReLU(),
nn.Linear(84, 10