网络搭建技巧

善用Sequential

pytorch网络搭建

首先将torch.nn构建的网络层转化为list类型,加[ ]
[conv , relu , norm]
将不同网络层list结合,相加+
list类型通过 * 操作可以提取列表
*self.model = nn.Sequential(model)

相关文章:

猜你喜欢
相关资源
相似解决方案