问如何在pytorch中初始化不同风格的nn.Sequential块的不同层的权重？
EN

Stack Overflow用户

提问于 2019-05-26 18:56:49

回答 2查看 1.4K关注 0票数 3

假设我有一个nn.Sequential块，它有两个线性层。我希望通过均匀分布来初始化第一层的权重，但希望将第二层的权重初始化为常量2.0。

net = nn.Sequential()
net.add_module('Linear_1', nn.Linear(2, 5, bias = False))
net.add_module('Linear_2', nn.Linear(5, 5, bias = False)

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/56312886

复制

相似问题

问如何在pytorch中初始化不同风格的nn.Sequential块的不同层的权重？EN