pytorch实现AlexNet
![在这里插入图片描述](https://img-blog.csdnimg.cn/f502b52ea6cf479397b1d9f66b804fdd.png)
def __init__(self):
super(AlexNet, self).__init__()
# 卷积部分
self.conv = nn.Sequential(
nn.Conv2d(1, 96, 11, 4), # in_channels, out_channels, kernel, stride, padding
nn.ReLU(),
nn.MaxPool2d(3, 2), # kernel, stride 55->27,channel = 95
nn.Conv2d(96, 256, 5, 1, 2), # 输入(27*27*96)
nn.ReLU(),
nn.MaxPool2d(3, 2),
nn.Conv2d(256, 384, 3, 1, 1),
nn.ReLU(),
nn.Conv2d(384, 384, 3, 1, 1),
nn.ReLU(),
nn.Conv2d(384, 256, 3, 1, 1),
nn.ReLU(),
nn.MaxPool2d(3, 2)
)
# 全连接部分
self.fc = nn.Sequential(
nn.Linear(256*5*5,4096),
nn.ReLU(),
nn.Dropo