pytorch学习笔记五:模型的创建

在上一节中整理了数据模块的知识点,在本节中主要围绕如何用pytorch构建一个模型来展开,最后用pytorch实现Alexnet网络结构的搭建。 下面基于上面的框架来探索每一个模块的实现细节。
一、模型的创建
上面是LetNet的网络图,由边和节点组成,节点表示输入的数据大小,而边就是数据之间的运算。从上面的LetNet的网络中可以看出,网络接受一个输入,然后经过运算得到一个输出,在网络结构的内部,又分为多个子网络层进行拼接组成,这些子网络层之间的拼接 配合,最终得到我们想要的输出。
所以通过上面的分析,可以得到构建模型的两大要素: ● 构建子模块(比如网络结构中的卷积层、池化层、激活层、全连接层); ● 拼接子模块(将子模块按照一定的顺序拼接起来,最终得到想到的网络结构)。
以人民币的二分类任务来构建LetNet网络结构:
class LeNet(nn.Module):
def __ini

pytorch学习笔记五:模型的创建最先出现在Python成神之路

版权声明:
作者:cc
链接:https://www.techfm.club/p/21251.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>