解決Pytorch中的神坑:關于model.eval的問題
更新時間：2021年05月22日 10:17:35 作者：汐夢聆海
這篇文章主要介紹了解決Pytorch中的神坑:關于model.eval的問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
有時候使用Pytorch訓練完模型，在測試數(shù)據(jù)上面得到的結果令人大跌眼鏡。
這個時候需要檢查一下定義的Model類中有沒有 BN 或 Dropout 層，如果有任何一個存在
那么在測試之前需要加入一行代碼：
#model是實例化的模型對象 model = model.eval()
表示將模型轉變?yōu)閑valuation（測試）模式，這樣就可以排除BN和Dropout對測試的干擾。
因為BN和Dropout在訓練和測試時是不同的：
對于BN，訓練時通常采用mini-batch，所以每一批中的mean和std大致是相同的；而測試階段往往是單個圖像的輸入，不存在mini-batch的概念。所以將model改為eval模式后，BN的參數(shù)固定，并采用之前訓練好的全局的mean和std；
對于Dropout，訓練階段，隱含層神經(jīng)元先乘概率P，再進行激活；而測試階段，神經(jīng)元先激活，每個隱含層神經(jīng)元的輸出再乘概率P。
如下圖所示：
補充：pytorch中model.eval之后是否還需要model.train的問題
答案是：需要的
正確的寫法是
for循環(huán)之后再開啟train，
循環(huán)之后的評估m(xù)odel.eval之后就會再次回到model.train
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Pytorch實現(xiàn)Fashion-mnist分類任務全過程
這篇文章主要介紹了Pytorch實現(xiàn)Fashion-mnist分類任務全過程，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-12-12
這篇文章主要給大家介紹了關于pandas如何刪除沒有列名的列的相關資料,需要的朋友可以參考下2021-11-11
python機器學習庫scikit-learn：SVR的基本應用
這篇文章主要介紹了python機器學習庫scikit-learn：SVR的基本應用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-06-06
這篇文章主要介紹了Python庫Gym開發(fā)和比較強化學習算法使用探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
這篇文章主要為大家詳細介紹了Mac版的Python、PyCharm安裝及使用方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-04-04
配置python連接oracle讀取excel數(shù)據(jù)寫入數(shù)據(jù)庫的操作流程
這篇文章主要介紹了配置python連接oracle，讀取excel數(shù)據(jù)寫入數(shù)據(jù)庫,本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
這篇文章主要給大家介紹了關于Django跨域請求問題解決的相關資料，文中介紹的實現(xiàn)方法包括：使用django-cors-headers全局控制、使用JsonP，只能用于Get方法以及在views.py里設置響應頭，只能控制單個接口，需要的朋友可以參考下2018-06-06
這篇文章主要介紹了Python一行命令部署http?ftp服務實現(xiàn)過程詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2024-01-01
這篇文章主要介紹了PyCharm設置注釋字體顏色以及是否傾斜的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-09-09
這篇文章主要介紹了Django框架視圖層URL映射與反向解析,結合實例形式分析了Django框架普通url映射、命名URL參數(shù)映射、分布式URL映射、反向解析等相關操作技巧,需要的朋友可以參考下2019-07-07