Pytorch實踐--eval 模式,每次預測相同的數據獲得的機率不一樣

描述:使用pytorch訓練好的模型對數據進行預測時,將model設置爲eval()模式以後,發現一樣的數據每次預測時的機率值會變更。web 分析:eval()能夠將模型切換爲 evaluation 模式,(僅僅當模型中含有Dropout和BatchNorm才起做用)。這裏所說的 「模型中含有Dropout和BatchNorm 」 是指 包含在*_modules* 中(當類成員的type派生於Mo
相關文章
相關標籤/搜索