直接原因:第一個Epoch訓練正常,第一次切換至Eval並進行測試正常,第二個Epochs訓練時報錯
根本原因:默認Model的模式是
net.train()
但是測試時需要調成
net.eval()
因為再次訓練時沒調回 train 模式,故出現上述 bug, 只需在訓練之前加上
net.train()
即可解決
直接原因:第一個Epoch訓練正常,第一次切換至Eval並進行測試正常,第二個Epochs訓練時報錯
根本原因:默認Model的模式是
net.train()
但是測試時需要調成
net.eval()
因為再次訓練時沒調回 train 模式,故出現上述 bug, 只需在訓練之前加上
net.train()
即可解決
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。