请问paddle中有没有什么接口可以将整个模型的梯度清零
收藏
请问paddle中有没有什么接口可以将整个模型的梯度清零,类型pytorch中的model.zero_grad()
0
收藏
请登录后评论
用优化器对象的 clear_grad() 方法,可以清楚此优化器优化的所有网络参数的梯度
文档里有介绍:
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训练模型”这部分。
感谢大佬帮助