pytorch 使用加載訓(xùn)練好的模型做inference
更新時間：2020年02月20日 15:41:28 作者：U_C
今天小編就為大家分享一篇pytorch 使用加載訓(xùn)練好的模型做inference，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
前提： 模型參數(shù)和結(jié)構(gòu)是分別保存的
1、 構(gòu)建模型(# load model graph)
model = MODEL()
2、加載模型參數(shù)(# load model state_dict)
model.load_state_dict
(
{
k.replace('module.',''):v for k,v in
torch.load(config.model_path, map_location=config.device).items()
}
)
model = self.model.to(config.device)
* config.device 指定使用哪塊GPU或者CPU
*k.replace('module.',''):v 防止torch.DataParallel訓(xùn)練的模型出現(xiàn)加載錯誤
（解決RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1問題）
3、設(shè)置當(dāng)前階段為inference（# predict）
model.eval()
以上這篇pytorch 使用加載訓(xùn)練好的模型做inference就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
NumPy實現(xiàn)ndarray多維數(shù)組操作
NumPy一個非常重要的作用就是可以進行多維數(shù)組的操作，這篇文章主要介紹了NumPy實現(xiàn)ndarray多維數(shù)組操作，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-05-05
Selenium是一種用于自動化網(wǎng)頁操作的工具,通過不同定位策略可以精準(zhǔn)定位網(wǎng)頁元素,本文介紹了8種定位方法,并詳細說明了每種方法的語法、使用場景及實際操作中的注意事項,需要的朋友可以參考下2025-02-02
Python使用Transformers實現(xiàn)機器翻譯功能
近年來,機器翻譯技術(shù)飛速發(fā)展,從傳統(tǒng)的基于規(guī)則的翻譯到統(tǒng)計機器翻譯,再到如今流行的神經(jīng)網(wǎng)絡(luò)翻譯模型,尤其是基于Transformer架構(gòu)的模型,翻譯效果已經(jīng)有了質(zhì)的飛躍,本文將詳細介紹如何使用Transformers庫來實現(xiàn)一個機器翻譯模型,需要的朋友可以參考下2024-11-11
這篇文章主要為大家詳細介紹了python編程實現(xiàn)歸并排序的具體代碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2017-04-04 
這篇文章主要介紹了python實現(xiàn)獲取序列中最小的幾個元素,是非常實用的技巧,需要的朋友可以參考下
2014-09-09 
這篇文章主要為大家詳細介紹了Python如何利用Sklearn實現(xiàn)異常檢測，文中的示例代碼講解詳細，對我們學(xué)習(xí)Python有一定的幫助，感興趣的可以跟隨小編一起學(xué)習(xí)一下
2022-12-12 
Python字符串中的單詞反轉(zhuǎn)的實現(xiàn)示例
在Python中,要將字符串中的單詞進行反轉(zhuǎn),本文主要介紹了Python字符串中的單詞反轉(zhuǎn)的實現(xiàn)示例,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2024-04-04 
關(guān)于python tushare Tkinter構(gòu)建的簡單股票可視化查詢系統(tǒng)（Beta v0.13）
這篇文章主要介紹了python tushare Tkinter構(gòu)建的簡單股票可視化查詢系統(tǒng)（Beta v0.13）,本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2020-10-10 
今天給大家?guī)淼氖顷P(guān)于Python的相關(guān)知識,文章圍繞著Python pygame繪制機制展開,文中有非常詳細的介紹及圖文示例,需要的朋友可以參考下
2021-06-06