在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]))

 

上一篇:cmuooc


下一篇:"ros::NodeHandle"的用法:全局vs.私有