记录一次Pytorch在定义模型优化时候没有参数

最新推荐文章于 2024-12-14 16:59:41 发布

十末.

最新推荐文章于 2024-12-14 16:59:41 发布

阅读量439

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_44646994/article/details/133909115

文章描述了在使用PyTorch的SGD优化器时遇到的ValueError，原因是模型类My_Model的`nn.Linear`层定义后多了一个逗号导致优化器没有接收到任何参数。解决方法是移除多余的逗号以正确设置模型结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    optimizer = torch.optim.SGD(model.parameters(),lr=0.001)
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Gancenty\.conda\envs\pytorch_cpu\Lib\site-packages\torch\optim\sgd.py", line 26, in __init__
    super().__init__(params, defaults)
  File "C:\Users\Gancenty\.conda\envs\pytorch_cpu\Lib\site-packages\torch\optim\optimizer.py", line 261, in __init__
    raise ValueError("optimizer got an empty parameter list")
ValueError: optimizer got an empty parameter list

报错现象： ValueError(“optimizer got an empty parameter list”)
报错原因：

class My_Model(nn.Module):
    def __init__(self, input_dim):
        super(My_Model, self).__init__()
        self.hidden_dim = 1
        self.layers=nn.Linear(input_dim, hidden_dim),
        
    def forward(self, x):
        x = self.layers(x)
        return x