用pytorch搭建AlexNet

Miracle
Miracle
Miracle
1357
文章
109
评论
2018年10月6日10:49:42深度学习用pytorch搭建AlexNet已关闭评论2,62531010字阅读3分22秒阅读模式

编译环境为Jupyter，代码如下：文章源自联网快讯-https://x1995.cn/3301.html

  1. #AlexNet 搭建
  2. import torch
  3. import torch.nn as nn
  4. class AlexNet(nn.Module):
  5.     def __init__(self):
  6.         super(AlexNet,self).__init__()
  7.         self.features=nn.Sequential(
  8.             nn.Conv2d(3,96,11,4,0),
  9.             nn.ReLU(True),
  10.             nn.MaxPool2d(3,2),
  12.             nn.Conv2d(96,256,5,1,2),
  13.             nn.ReLU(True),
  14.             nn.MaxPool2d(3,2),
  16.             nn.Conv2d(256,384,3,1,1),
  17.             nn.ReLU(True),
  19.             nn.Conv2d(384,384,3,1,1),
  20.             nn.ReLU(True),
  22.             nn.Conv2d(384,256,3,1,1),
  23.             nn.ReLU(True),
  24.             nn.MaxPool2d(3,2)
  25.         )
  26.         self.classifier=nn.Sequential(
  27.             nn.Linear(9216,4096),
  28.             nn.ReLU(True),
  29.             nn.Dropout(0.5),
  31.             nn.Linear(4096,4096),
  32.             nn.ReLU(True),
  33.             nn.Dropout(0.5),
  35.             nn.Linear(4096,1000)
  36.         )
  37.     def forward(self,x):
  38.         x=self.features(x)
  39.         x=x.view(x.size(0),-1)
  40.         x=self.classifier(x)
  41.         return x
  42. alexnet=AlexNet()
  43. print(alexnet)

运行后网络结构：
用pytorch搭建AlexNet文章源自联网快讯-https://x1995.cn/3301.html

继续阅读
历史上的今天
10月
6
Miracle
  • 本文由 发表于 2018年10月6日10:49:42