利用autorgrad实现线性回归

教程大全  / 倒序浏览   ©

#楼主# 2020-2-16

跳转到指定楼层

马上注册,分享更多源码,享用更多功能,让你轻松玩转云大陆。

您需要 登录 才可以下载或查看,没有帐号?立即注册

x

利用pytorch的主动求导机制实现线性回归实在非常简单,但是在实现过程中犯了一个错误调试了很久,需要留意的是在更新梯度时候的写法,如下代码所示:

[	DISCUZ_CODE_0	]

其中错误写法导致了梯度发生了变化,错误写法导致了梯度变成了lr,然后在计算图中传递下去,如下是完成的正确实现代码:

1. 天生数据

[	DISCUZ_CODE_1	]

2. 算法实现

[	DISCUZ_CODE_2	]
分享淘帖
回复

使用道具

您的回复是对作者最大的奖励

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

关于作者

  • 主题

    7

  • 帖子

    7

  • 关注者

    0

Archiver|手机版|小黑屋|云大陆 | 赣ICP备18008958号-4|网站地图
Powered by vrarz.com!  © 2019-2020版权所有云大陆