TypeError: ‘method‘ object is not iterable
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-pa
·
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)
更多推荐
已为社区贡献1条内容
所有评论(0)