这是预期的 - 存储模块
list
,
dict
,
set
或其他python容器没有注册它们所拥有的模块
list
等。要使代码正常工作,请使用
nn.ModuleList
代替。它就像修改你的一样简单
init
要使用的代码
layers = []
new_input_size = self.input_size
for i in xrange(num_layers):
layers.append(LSTMCell(new_input_size, hidden_size))
new_input_size = hidden_size
self.layers = nn.ModuleList(layers)
</code>