💩

【PyTorch】TypeError: 'method' object is not iterableを解決

2022/08/21に公開

多分PyTorchに限った問題ではないと思うのですが、
残しておきます。

違いは以下だけです。

optimizer = torch.optim.SGD(params = model.parameters, lr = 0.01)
# type(model.parameters) ->  <class 'method'>

optimizer = torch.optim.SGD(params = model.parameters(), lr = 0.01)
# type(model.parameters) ->  <class 'generator'>

parametersはメソッドなので、実行()をつけてやらないと、メソッドそのものを渡していることになります。
これでは、確かにイテラブルではありません。
generaterならイテラブルなのでOKというわけでした〜。

Discussion