model.zero_grad() & optimizer.zero_grad()


 有兩種方式把模型的參數梯度設成0:

model.zero_grad()
optimizer.zero_grad()#當optimizer=optim.Optimizer(model.parameters())時,兩者等效

 如果只想要把某一Variable的梯度置為0,只需要以下語句:

Variable.grad.data.zero_()

 


參考1:model.zero_grad() 與 optimizer.zero_grad()


 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM