💩
【PyTorch】TypeError: 'method' object is not iterableを解決
多分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