ValueError：优化器得到一个空的参数列表

Question

我创建了以下简单的线性类：

class Decoder(nn.Module):
    def __init__(self, K, h=()):
        super().__init__()
        h = (K,)+h+(K,)
        self.layers = [nn.Linear(h1,h2) for h1,h2 in zip(h, h[1:])]

    def forward(self, x):
        for layer in self.layers[:-1]:
            x = F.relu(layer(x))
        return self.layers[-1](x)

但是，当我尝试将参数放入优化器类时，我收到错误 ValueError: optimizer got an empty parameter list。

decoder = Decoder(4)
LR = 1e-3
opt = optim.Adam(decoder.parameters(), lr=LR)

类定义有什么明显错误的地方吗？

Answer 1

由于您将层存储在 Decoder 内的常规 pythonic 列表中，Pytorch 无法告诉 self.list 的这些成员实际上是子模块。把这个列表转换成pytorch的nn.ModuleList，你的问题就解决了

class Decoder(nn.Module):
    def __init__(self, K, h=()):
        super().__init__()
        h = (K,)+h+(K,)
        self.layers = nn.ModuleList(nn.Linear(h1,h2) for h1,h2 in zip(h, h[1:]))