Torch.no_grad()影響MSE損失


相關描述

https://discuss.pytorch.org/t/torch-no-grad-affecting-outputs-loss/28595/3

今天在訓練網絡的時候,發現mseloss在train和test時相差了好幾個量級,后來發現是因為mseloss(input,target),並不是對稱的,input的應該是網絡的輸出,而targte是目標,如果顛倒過來,計算就是錯的。


免責聲明!

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



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