在数据模块中我们了解了pytorch是如何从硬盘中读取数据,然后对数据进行预处理,最后转化为张量的形式输入到我们的模型当中。
在模型里,是对数据模块出处的张量进行复杂的数学运算,最后得到可以分类、分割、目标检测的输出。
一、网络模型创建的步骤这一节学习模型模块。
模型模块又分两个模块,一个是模型创建,一个是权值初始化。 在模型创建中,又分为两个子模块。构建网络层和拼接网络层。 构建网络层,例如卷积层、池化层、激活函数层等。 构建好网络层之后,再按照一定的顺序进行拼接,得到复杂的神经网络。
创建好模型之后,我们要对模型的权值进行初始化。pytorch提供了丰富的初始化方法。。
构建子模块,是在__int__()中实现的。 拼接子模块,是在froward()中实现的。
举例: |
网络模型创建的步骤及nn.Module属性
最新推荐文章于 2024-07-27 17:42:19 发布
本文详细介绍了如何创建深度学习的网络模型,包括利用nn.Module设计自定义模块、初始化权重、构建前向传播流程等关键步骤。通过实例解析,帮助读者理解并掌握网络模型创建的全过程。
摘要由CSDN通过智能技术生成