请选择 进入手机版 | 继续访问电脑版
查看: 107|回复: 3

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

[复制链接]

27

主题

37

帖子

116

积分

论坛管理

Rank: 4

积分
116
发表于 2018-10-15 18:10:12 | 显示全部楼层 |阅读模式
如果你有一个很多层的模型,你可以先创建一个列表然后使用*运算符将列表展开为位置参数,如下所示:
  1. layers = []
  2. layers.append(nn.Linear(3, 4))
  3. layers.append(nn.Sigmoid())
  4. layers.append(nn.Linear(4, 1))
  5. layers.append(nn.Sigmoid())

  6. net = nn.Sequential(*layers)
复制代码
  1. net = nn.Sequential(
  2.       nn.Linear(3, 4),
  3.       nn.Sigmoid(),
  4.       nn.Linear(4, 1),
  5.       nn.Sigmoid()
  6.       ).to(device)


  7. print(net)

  8. Sequential(
  9.   (0): Linear(in_features=3, out_features=4, bias=True)
  10.   (1): Sigmoid()
  11.   (2): Linear(in_features=4, out_features=1, bias=True)
  12.   (3): Sigmoid()
  13.   )
复制代码
代码的结构类似keras,直接添加。
回复

使用道具 举报

326

主题

666

帖子

4179

积分

xdtech

Rank: 5Rank: 5

积分
4179
发表于 2019-2-24 20:29:12 | 显示全部楼层
代码还是
比较友好的

回复

使用道具 举报

326

主题

666

帖子

4179

积分

xdtech

Rank: 5Rank: 5

积分
4179
发表于 2019-2-24 20:29:26 | 显示全部楼层
封装的比较严重,
灵活性,好不好呢
回复

使用道具 举报

326

主题

666

帖子

4179

积分

xdtech

Rank: 5Rank: 5

积分
4179
发表于 2019-2-24 20:29:36 | 显示全部楼层
顺序模型,
主要在哪里用?
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表