optimizer.zero_grad() ,用于将模型的参数梯度初始化为0。
#反向计算 loss.backward()#反向传播计算梯度 optimizer.step()#更新参数,梯度被反向计算之后,调用函数进行所有参数更新 #在反向传播计算时,梯度的计算是累加,但是每个batch处理时不需要和其它batch混合起来累加计算,所以对每个batch调用zero_grad将参数梯度置0 #如果不是每个batch清理依次梯度,而是两次或者多次清理,相当于提高了batch批次大小 optimizer.zero_grad()#将模型参数梯度置为0;
本站资源均来自互联网,仅供研究学习,禁止违法使用和商用,产生法律纠纷本站概不负责!如果侵犯了您的权益请与我们联系!
转载请注明出处: 免费源码网-免费的源码资源网站 » pytorch中的zero_grad()函数的含义和使用
发表评论 取消回复