Question:
```handlebars
Traceback (most recent call last):
File "G:/pycharm Project/PytorchLearning/first/Classifier.py", line 210, in <module>
main()
File "G:/pycharm Project/PytorchLearning/first/Classifier.py", line 164, in main
optimizer = optim.SGD(net.parameters, lr=0.001, momentum=0.9)
File "D:\Anaconda\envs\gpu-env\lib\site-packages\torch\optim\sgd.py", line 64, in __init__
super(SGD, self).__init__(params, defaults)
File "D:\Anaconda\envs\gpu-env\lib\site-packages\torch\optim\optimizer.py", line 44, in __init__
param_groups = list(params)
TypeError: 'method' object is not iterable
原代码:
net = Net()
#
# # lr: 学习率
optimizer = optim.SGD(net.parameters, lr=0.001, momentum=0.9)
#
print(optimizer)
Solution:
修改后:
net = Net()
#
# # lr: 学习率
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
#
print(optimizer)
问题所在:
net.parameters后面没加()
。 😢…骂骂咧咧