Как создать Pytorch NN с 2 скрытых слоя с nn.Sequential?

Question

Как добавить еще один скрытый слой в этой модели 1 слоя? :

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

Answer 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)) ]))

Единственное, что вам нужно сделать, - это взять 1-й скрытый слой (H1) в качестве входных данных для Следующий линейный слой, который будет выводить на другой скрытый слой (H2), затем мы добавим еще один слой активации Tanh и, наконец, добавим линейный слой, который принимает слой H2 в качестве входных данных и выводит количество выходных узлов.