关于关于pytorch中全连接神经网络搭建两种模式详解中全连接神经网络搭建两种模式详解
今天小编就为大家分享一篇关于pytorch中全连接神经网络搭建两种模式详解，具有很好的参考价值，希望对大
家有所帮助。一起跟随小编过来看看吧
pytorch搭建神经网络是很简单明了的，这里介绍两种自己常用的搭建模式：
import torch
import torch.nn as nn
first：：
class NN(nn.Module):
def __init__(self):
super(NN,self).__init__()
self.model=nn.Sequential(
nn.Linear(30,40),
nn.ReLU(),
nn.Linear(40,60),
nn.Tanh(),
nn.Linear(60,10),
nn.Softmax()
)
self.model[0].weight.data.uniform_(-3e-3, 3e-3)
self.model[0].bias.data.uniform(-1,1)
def forward(self,states):
return self.model(states)
这一种是将整个网络写在一个Sequential中，网络参数设置可以在网络搭建好后单独设置：
self.model[0].weight.data.uniform_(-3e-3,3e-3)，这是设置第一个linear的权重是（-3e-3，3e-3）之间的均匀分布,bias是-1至1
之间的均匀分布。
second：：
class NN1(nn.Module):
def __init__(self):
super(NN1,self).__init__()
self.Linear1=nn.Linear(30,40)
self.Linear1.weight.data.fill_(-0.1)
#self.Linear1.weight.data.uniform_(-3e-3,3e-3)
self.Linear1.bias.data.fill_(-0.1)
self.layer1=nn.Sequential(self.Linear1,nn.ReLU())
self.Linear2=nn.Linear(40,60)
self.layer2=nn.Sequential(self.Linear2,nn.Tanh())
self.Linear3=nn.Linear(60,10)
self.layer3=nn.Sequential(self.Linear3,nn.Softmax())
def forward(self,states):
return self.model(states)
网络参数的设置可以在定义完线性层之后直接设置如这里对于第一个线性层是这样设置：self.Linear1.weight.data.fill_(-
0.1),self.Linear1.bias.data.fill_(-0.1)。
你可以看一下这样定义完的参数的效果：你可以看一下这样定义完的参数的效果：
Net=NN()
print("0:",Net.model[0])
print("weight:",type(Net.model[0].weight))
print("weight:",type(Net.model[0].weight.data))
print("bias",Net.model[0].bias.data)
print('1:',Net.model[1])
#print("weight:",Net.model[1].weight.data)
print('2:',Net.model[2])
print('3:',Net.model[3])
#print(Net.model[-1])
Net1=NN1()
print(Net1.Linear1.weight.data)
输出：
0: Linear (30 -> 40)
weight: <class 'torch.nn.parameter.Parameter'>
weixin_38516863
- 粉丝: 3
- 资源: 970
我的内容管理
收起
- 我的资源
快来上传第一个资源
我的收益 登录查看自己的收益
我的积分
登录查看自己的积分
我的C币
登录后查看C币余额
我的收藏 
我的下载 
下载帮助
会员权益专享
最新资源
- ARM Cortex-A(armV7)编程手册V4.0.pdf
- ABB机器人保养总结解析.ppt
- 【超详细图解】菜鸡如何理解双向链表的python代码实现
- 常用网络命令的使用 ipconfig ping ARP FTP Netstat Route Tftp Tracert Telnet nslookup
- 基于单片机控制的DC-DC变换电路
- RS-232接口电路的ESD保护.pdf
- linux下用time(NULL)函数和localtime()获取当前时间的方法
- Openstack用户使用手册.docx
- KUKA KR 30 hA,KR 60 hA机器人产品手册.pdf
- Java programming with JNI
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
评论0