易学智能论坛 › PyTorch › 如何编写一个pytorch顺序模型？

fantomas 发表于 2018-10-15 18:10:12

如何编写一个pytorch顺序模型？

如果你有一个很多层的模型，你可以先创建一个列表然后使用*运算符将列表展开为位置参数，如下所示：
layers = []
layers.append(nn.Linear(3, 4))
layers.append(nn.Sigmoid())
layers.append(nn.Linear(4, 1))
layers.append(nn.Sigmoid())

net = nn.Sequential(*layers)net = nn.Sequential(
      nn.Linear(3, 4),
      nn.Sigmoid(),
      nn.Linear(4, 1),
      nn.Sigmoid()
      ).to(device)


print(net)

Sequential(
(0): Linear(in_features=3, out_features=4, bias=True)
(1): Sigmoid()
(2): Linear(in_features=4, out_features=1, bias=True)
(3): Sigmoid()
)代码的结构类似keras，直接添加。

shaoheshaohe 发表于 2019-2-24 20:29:12

代码还是
比较友好的

shaoheshaohe 发表于 2019-2-24 20:29:26

封装的比较严重，
灵活性，好不好呢

shaoheshaohe 发表于 2019-2-24 20:29:36

顺序模型，
主要在哪里用？

查看完整版本: 如何编写一个pytorch顺序模型？