AlexNet学习笔记(再次进行cifar10数据分类)

网络结构

网络结构如图所示，包含5个卷积层和3个全连接层。其中，有3个卷积层进行了最大池化。

AlexNet各层组织结构如表所示，其中conv表示卷积运算操作，ReLU表示校正线性单元，pool表示池化操作，norm表示局部响应归一化，dropout表示丢失输出操作，IP表示全连接。

根据AlexNet结构搭建网络，代码如下：

from torch import nn

class AlexNet(nn.Module):
    def __init__(self):
        super().__init__()

        self.conv1 = nn.Sequential(
            # nn.Conv2d(3, 96,kernel_size=11,stride=4),
            nn.Conv2d(3, 6,
                      kernel_size=3,
                      stride=1),
            nn.ReLU(),
            nn.MaxPool2d(