Pytorch 线性回归 grad清零报错：w.grad.data.zero_() AttributeError: 'NoneType' object has no attribute 'data'

最新推荐文章于 2024-12-27 20:10:58 发布

刺无名客

最新推荐文章于 2024-12-27 20:10:58 发布

阅读量1.1w

点赞数 8

分类专栏：处理问题文章标签： python 线性回归 Pytorh

本文链接：https://blog.youkuaiyun.com/m0_37637704/article/details/101019438

版权

在学习Pytorch线性回归时遇到AttributeError：'NoneType' object has no attribute 'data'的问题。错误源于尝试对None类型的变量调用.data.zero_()。解决方法包括确保在for循环最后进行梯度清零，或者在循环前先进行一次训练。教程中的loss打印也需要修正为loss.item()。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

学习了https://github.com/L1aoXingyu/code-of-learn-deep-learning-with-pytorch/blob/master/chapter3_NN/linear-regression-gradient-descend.ipynb 上的Pytorch线性回归这一章，自己复了一遍代码后，再删去了一些人为学习型的输出和plot画图代码后报错

w.grad.data.zero_()

AttributeError: 'NoneType' object has no attribute 'data'

很明显。问题出在梯度清零上。

以下是错误代码

import torch
import numpy as np
from torch.autograd import Variable
import matplotlib.pyplot as plt

torch.manual_seed(2017)


w_target = np.array([0.5, 3, 2.4]) # 定义参数
b_target = np.array([0.9]) # 定义参数

x_sample=np.arange(-3,3.1,0.1)
y_sample=b_target[0] + w_target[0] *x_sample+w_target[1]*x_sample**2 +w_target[2]*x_sample**3


x_train=np.stack([x_sample ** i for i in range(1,4)],axis&#