File "/opt/asr/notebook/torch_learn/nn_example.py", line 42, in <module>
    optimzer = optim.Adam(params = model.parameters, lr=learning_rate)
  File "/opt/anaconda3/envs/notebook/lib/python3.8/site-packages/torch/optim/adam.py", line 48, in __init__
    super(Adam, self).__init__(params, defaults)
  File "/opt/anaconda3/envs/notebook/lib/python3.8/site-packages/torch/optim/optimizer.py", line 48, in __init__
    param_groups = list(params)
TypeError: 'method' object is not iterable

报错代码
optimzer = optim.Adam(params = model.parameters, lr=learning_rate)

model的参数是方法不是属性
def parameters(self, recurse: bool = True) -> Iterator[Parameter]:

正确写法
optimzer = optim.Adam(params = model.parameters(), lr=learning_rate)

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐