博客主页瑕疵的CSDN主页 Gitee主页瑕疵的gitee主页⏩ 文章专栏《热点资讯》被SGD坑了参数传错的深夜血泪史目录昨晚写CNN模型SGD优化器突然报错。我盯着屏幕心想学习率0.01应该没问题啊结果弹出报错TypeError: SGD() argument params must be an iterable of Tensors, but got class torch.nn.Module。核心根源SGD优化器要的是参数列表不是模型本身。我直接传了modelPyTorch直接崩了。model.parameters()才是返回可训练参数的正确方式。我测试过10次错误原因就这一个——新手总以为传模型就行。错误示范我踩过的坑# 错误直接传入model不是参数optimizeroptim.SGD(model,lr0.01)# 运行就报错# 为什么model是整个神经网络模块不是参数列表正确姿势改完直接跑通# 正确必须传model.parameters()optimizeroptim.SGD(model.parameters(),lr0.01)# 重点# model.parameters()返回可训练参数的迭代器截图里清楚显示got
PyTorch SGD优化器报错怎么办？教你一招避坑
发布时间：2026/6/9 21:04:05
博客主页瑕疵的CSDN主页 Gitee主页瑕疵的gitee主页⏩ 文章专栏《热点资讯》被SGD坑了参数传错的深夜血泪史目录昨晚写CNN模型SGD优化器突然报错。我盯着屏幕心想学习率0.01应该没问题啊结果弹出报错TypeError: SGD() argument params must be an iterable of Tensors, but got class torch.nn.Module。核心根源SGD优化器要的是参数列表不是模型本身。我直接传了modelPyTorch直接崩了。model.parameters()才是返回可训练参数的正确方式。我测试过10次错误原因就这一个——新手总以为传模型就行。错误示范我踩过的坑# 错误直接传入model不是参数optimizeroptim.SGD(model,lr0.01)# 运行就报错# 为什么model是整个神经网络模块不是参数列表正确姿势改完直接跑通# 正确必须传model.parameters()optimizeroptim.SGD(model.parameters(),lr0.01)# 重点# model.parameters()返回可训练参数的迭代器截图里清楚显示got
