如何使用 nn.Sequential 创建具有 2 个隐藏层的 pytorch NN？

【问题标题】：how to create a pytorch NN with 2 hidden layer with nn.Sequential?如何使用 nn.Sequential 创建具有 2 个隐藏层的 pytorch NN？
【发布时间】：2020-05-11 23:35:59
【问题描述】：

如何在这个 1 层模型中添加另一个隐藏层？：

model = nn.Sequential(OrderedDict([ ('fc1', nn.Linear(D_in, H)), 
                                    ('Tanh', nn.Tanh()),
                                    ('fc2', nn.Linear(H, D_out))]))

【问题讨论】：

标签： python-3.x pytorch

【解决方案1】：

您可以通过将原始隐藏层一分为二或添加另一个隐藏层来做到这一点：

model = nn.Sequential(OrderedDict([ ('fc1', nn.Linear(D_in, H1)),
                                    ('act1', nn.Tanh()),
                                    ('fc2', nn.Linear(H1, H2)),
                                    ('act2', nn.Tanh()),
                                    ('fc3', nn.Linear(H2, D_out)) ]))

您唯一要做的就是将第一个隐藏层 (H1) 作为输入到下一个线性层，该线性层将输出到另一个隐藏层 (H2)，然后我们添加另一个 Tanh 激活层，最后，我们添加一个以H2层为输入，输出为输出节点数的线性层。

【讨论】：