网络结构
网络结构如图所示,包含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(