如何使用飞桨的自动求导完成线性回归?
收藏
尝试使用paddle完成动手学深度学习里的内容,在实现简易的线性回归模型中,实现小批量梯度下降时出错,错误信息
:AttributeError: 'NoneType' object has no attribute 'zero_'
错误代码如下:
import paddle def sgd(params,lr,batch_size): with paddle.no_grad(): for param in params: param -= lr*param.grad/batch_size param.grad.zero_() lr =0.3 num_epochs=3 net=linreg loss=squared_loss for epoch in range(num_epochs): for X,y in data_iter(batch_size,features,labels): l=loss(net(X,w,b),y).sum() l.backward() sgd([w,b],lr,10) print(w,b)
0
收藏
请登录后评论
grad里面没有这个方法,建议你查看文档:https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/grad_cn.html#grad
公开项目里也有自己写线性回归的,可以找找看