pytorch之Sequential 和 Module
2019-03-26 本文已影响0人
很菜_嗯
在构建比较大的网络层结构时,比如100层网络结构,这个时候再去手动定义参数就显得非常麻烦,所以 PyTorch 提供了两个模块来帮助我们构建模型,一个是Sequential,一个是 Module。
Sequential 允许我们构建序列化的模块,而 Module 是一种更加灵活的模型定义方式。
Sequential模型
Module
将整个网络层的结构操作图表达出来,即前馈神经网络。在__init__中定义每层的结构,在forward中定义前向传播的路径。