其他分享
首页 > 其他分享> > 在Pytorch中使用 nn.Sequential()自动添加网络层

在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