错误：优化器得到一个空的参数列表。 如何修改我的代码？

Question

这是代码：

class MLP(nn.Module):  
    def __ini__(self):
         super (MLP, self). __init__()  
         self.model=nn.nn.Sequential(  
            nn.Linear(784, 200),
            nn.LeakyReLU(inplace=True),  
            nn.Linear(200, 200),
            nn.LeakyReLU(inplace=True),
            nn.Linear(200, 10),
            nn.LeakyReLU(inplace=True),
            )

    def forward(self,x):
                x=self.model(x)
                return 
device= torch.device('cuda:0')
net = MLP().to(device)

运行这些代码时

optimizer = optim.SGD(net.parameters(),lr=learning_rate)

我明白了

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

我试图模仿这个笔记本。

Answer 1

您的代码有几个问题。 目前您的代码相当于：

class MLP(nn.Module):
    def forward(self, x):
        pass

以下是问题：

1. 您的初始化程序名为__init__ 。

2. forward应该返回一个值，这里是x 。

3. 将nn.nn.Sequential替换为nn.Sequential 。

---

class MLP(nn.Module):
    def __init__(self):
        super (MLP, self).__init__()  
        self.model = nn.Sequential(  
            nn.Linear(784, 200),
            nn.LeakyReLU(inplace=True),  
            nn.Linear(200, 200),
            nn.LeakyReLU(inplace=True),
            nn.Linear(200, 10),
            nn.LeakyReLU(inplace=True))

    def forward(self,x):
        x = self.model(x)
        return x

device= torch.device('cuda:0')
net = MLP().to(device)

optimizer = torch.optim.SGD(net.parameters(), lr=learning_rate)