Pytorch没有显示LSTMCell参数

作者: =_=
发布时间: 2025-02-06 01:15:55 (26天前)
转自：

2 条回复

0#
回复此人
甲基蓝 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 这是预期的 - 存储模块 <code> list </code> ， <code> dict </code> ， <code> set </code> 或其他python容器没有注册它们所拥有的模块 <code> list </code> 等。要使代码正常工作，请使用 <a href="https://pytorch.org/docs/master/nn.html#modulelist" rel="nofollow noreferrer"> <code> nn.ModuleList </code> </A> 代替。它就像修改你的一样简单 <code> __init__ </code> 要使用的代码 </p> <pre> <code> 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> </pre> </DIV>

编辑

登录后才能参与评论