首页 Paddle框架 帖子详情
请问paddle中有没有什么接口可以将整个模型的梯度清零
收藏
快速回复
Paddle框架 问答模型训练 1283 3
请问paddle中有没有什么接口可以将整个模型的梯度清零
收藏
快速回复
Paddle框架 问答模型训练 1283 3

请问paddle中有没有什么接口可以将整个模型的梯度清零,类型pytorch中的model.zero_grad()

0
收藏
回复
全部评论(3)
时间顺序
FutureSI
#2 回复于2021-09

用优化器对象的 clear_grad() 方法,可以清楚此优化器优化的所有网络参数的梯度

0
回复
FutureSI
#3 回复于2021-09

文档里有介绍:

https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/02_paddle2.0_develop/05_train_eval_predict_cn.html

参考“3.1 拆解Model.prepare()、Model.fit()-- 用基础API训练模型”这部分。

0
回复
白告
#4 回复于2021-09

感谢大佬帮助

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户