网络模型创建的步骤及nn.Module属性

本文详细介绍了如何创建深度学习的网络模型,包括利用nn.Module设计自定义模块、初始化权重、构建前向传播流程等关键步骤。通过实例解析,帮助读者理解并掌握网络模型创建的全过程。
摘要由CSDN通过智能技术生成

 

在数据模块中我们了解了pytorch是如何从硬盘中读取数据,然后对数据进行预处理,最后转化为张量的形式输入到我们的模型当中。

 

在模型里,是对数据模块出处的张量进行复杂的数学运算,最后得到可以分类、分割、目标检测的输出。

 

 

一、网络模型创建的步骤

这一节学习模型模块。

 

 

模型模块又分两个模块,一个是模型创建,一个是权值初始化。

在模型创建中,又分为两个子模块。构建网络层和拼接网络层。

构建网络层,例如卷积层、池化层、激活函数层等。

构建好网络层之后,再按照一定的顺序进行拼接,得到复杂的神经网络。

 

创建好模型之后,我们要对模型的权值进行初始化。pytorch提供了丰富的初始化方法。。

 

构建子模块,是在__int__()中实现的。

拼接子模块,是在froward()中实现的。

 

 

 

举例:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值