Pytorch没有显示LSTMCell参数


=_=
2024-12-11 07:54:16 (1月前)


我有以下代码:

class myLSTM(nn.Module):
def init (self,inputsize,output_size,hidden​​size,num_layers):
super(myLSTM,self).
init
()
self.input_size = input_size + 1


2 条回复
  1. 0# 甲基蓝 | 2019-08-31 10-32



    这是预期的 - 存储模块

    list



    dict



    set

    或其他python容器没有注册它们所拥有的模块

    list

    等。要使代码正常工作,请使用


    nn.ModuleList


    代替。它就像修改你的一样简单

    init

    要使用的代码




    1. 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)

    2. </code>

登录 后才能参与评论