在Pytorch中使用 nn.Sequential()自动添加网络层
作者:互联网
nn.Sequential()定义网络简单高效,可以写死,可以自动添加add_module
参考链接:pytorch中的add_module函数 - 蒙面的普罗米修斯 - 博客园 (cnblogs.com)
pytorch nn.Sequential()动态添加方法 - 慢行厚积 - 博客园 (cnblogs.com)
self.encoder = nn.Sequential() for i in range(len(self.nH_enc)): if i == 0: self.encoder.add_module('layer_{}'.format(i), nn.Linear(self.input_dim, self.nH_enc[i])) else: self.encoder.add_module('layer_{}'.format(i), nn.Linear(self.nH_enc[i-1], self.nH_enc[i]))
标签:nH,nn,self,add,module,Pytorch,Sequential 来源: https://www.cnblogs.com/vvzhang/p/14991127.html