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