pytorch网络搭建

网络搭建技巧 善用Sequential 善用Sequential 首先将torch.nn构建的网络层转化为list类型,加[ ] [conv , relu , norm] 将不同网络层list结合,相加+ list类型通过 * 操作可以提取列表 *self.model = nn.Sequential(model)
相关文章
相关标签/搜索