使用Pytorch訓練,報錯Runtime Error: cudnn RNN backward can only be called in training mode


直接原因:第一個Epoch訓練正常,第一次切換至Eval並進行測試正常,第二個Epochs訓練時報錯

根本原因:默認Model的模式是

net.train()

但是測試時需要調成

net.eval()

因為再次訓練時沒調回 train 模式,故出現上述 bug, 只需在訓練之前加上

net.train()

即可解決


免責聲明!

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



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