Pytorch的訓練模式和測試模式切換
由於訓練的時候Dropout和BN層起作用，每個batch BN層的參數不一樣，dropout在訓練時隨機失效點具有隨機性，所以訓練和測試要區分開來。
使用時切記要根據實際情況切換：
model.train()
model.eval()
切記！！！！！！！！
究竟錯幾遍才能記住呢？讓我們拭目以待！ -.-
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。