学习了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&#