pytorch預測之解決多次預測結果不一致問題
更新時間：2021年06月01日 08:40:15 作者：confusingbird
這篇文章主要介紹了pytorch多次預測結果不一致的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
為什么多次預測結果不一致
1、檢查是否在每次預測前使用
model.eval()
或者是
with torch.no_grad(): for ...
推薦下面的方法，上面的的方法計算梯度，但是并不反向傳播，下面的方法既不計算梯度，也不反向傳播，速度更快。
2、檢查是否取消了所有的dropout
3、設置隨機種子
def setup_seed(seed):
np.random.seed(seed)
random.seed(seed)
torch.manual_seed(seed) #cpu
torch.cuda.manual_seed_all(seed) #并行gpu
torch.backends.cudnn.deterministic = True #cpu/gpu結果一致
torch.backends.cudnn.benchmark = True #訓練集變化不大時使訓練加速
4、保證實例化模型前要將is_training置為false；這兩行代碼順序不能顛倒
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Pytorch中TensorDataset與DataLoader的使用方式
這篇文章主要介紹了Pytorch中TensorDataset與DataLoader的使用方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09
python gui開發(fā)——制作抖音無水印視頻下載工具(附源碼)
這篇文章主要介紹了python gui開發(fā)——制作抖音無水印視頻下載工具(附源碼)的相關資料，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-02-02
這篇文章主要為大家詳細介紹了python實現(xiàn)簡單成績錄入系統(tǒng)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-09-09
Python操作Sonqube API獲取檢測結果并打印過程解析
這篇文章主要介紹了Python操作Sonqube API獲取檢測結果并打印過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-11-11
Python使用xpath對解析內容進行數(shù)據(jù)提取
XPath 使用路徑表達式來選取HTML/ XML 文檔中的節(jié)點或節(jié)點集,節(jié)點是通過沿著路徑 (path) 或者步 (steps) 來選取的,本文將給大家介紹Python使用xpath對解析內容進行數(shù)據(jù)提取的方法,需要的朋友可以參考下2024-05-05
python創(chuàng)建只讀屬性對象的方法(ReadOnlyObject)
有時需要創(chuàng)建一個帶只讀屬性的對象，大家可以參考下如下的方法進行創(chuàng)建，稍加改造，可以得到很特殊的效果2013-02-02
python 如何將數(shù)據(jù)寫入本地txt文本文件的實現(xiàn)方法
這篇文章主要介紹了python 如何將數(shù)據(jù)寫入本地txt文本文件的實現(xiàn)方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-09-09
今天小編就為大家分享一篇關于理想高通濾波實現(xiàn)Python opencv示例，小編覺得內容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-01-01
python和C++一樣,支持多繼承,本文主要介紹了Python類的多繼承的搜索順序,文中通過示例代碼介紹的非常詳細,需要的朋友們下面隨著小編來一起學習學習吧2024-03-03
這篇文章主要介紹了python爬蟲刷訪問量 2019 7月，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08