pytorch 使用加載訓(xùn)練好的模型做inference
更新時(shí)間：2020年02月20日 15:41:28 作者：U_C
今天小編就為大家分享一篇pytorch 使用加載訓(xùn)練好的模型做inference，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
前提： 模型參數(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)加載錯(cuò)誤
（解決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)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
NumPy實(shí)現(xiàn)ndarray多維數(shù)組操作
NumPy一個(gè)非常重要的作用就是可以進(jìn)行多維數(shù)組的操作，這篇文章主要介紹了NumPy實(shí)現(xiàn)ndarray多維數(shù)組操作，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-05-05
Selenium是一種用于自動(dòng)化網(wǎng)頁操作的工具,通過不同定位策略可以精準(zhǔn)定位網(wǎng)頁元素,本文介紹了8種定位方法,并詳細(xì)說明了每種方法的語法、使用場景及實(shí)際操作中的注意事項(xiàng),需要的朋友可以參考下2025-02-02
Python使用Transformers實(shí)現(xiàn)機(jī)器翻譯功能
近年來,機(jī)器翻譯技術(shù)飛速發(fā)展,從傳統(tǒng)的基于規(guī)則的翻譯到統(tǒng)計(jì)機(jī)器翻譯,再到如今流行的神經(jīng)網(wǎng)絡(luò)翻譯模型,尤其是基于Transformer架構(gòu)的模型,翻譯效果已經(jīng)有了質(zhì)的飛躍,本文將詳細(xì)介紹如何使用Transformers庫來實(shí)現(xiàn)一個(gè)機(jī)器翻譯模型,需要的朋友可以參考下2024-11-11
這篇文章主要為大家詳細(xì)介紹了python編程實(shí)現(xiàn)歸并排序的具體代碼，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2017-04-04 
python實(shí)現(xiàn)獲取序列中最小的幾個(gè)元素
這篇文章主要介紹了python實(shí)現(xiàn)獲取序列中最小的幾個(gè)元素,是非常實(shí)用的技巧,需要的朋友可以參考下
2014-09-09 
Python+Sklearn實(shí)現(xiàn)異常檢測
這篇文章主要為大家詳細(xì)介紹了Python如何利用Sklearn實(shí)現(xiàn)異常檢測，文中的示例代碼講解詳細(xì)，對(duì)我們學(xué)習(xí)Python有一定的幫助，感興趣的可以跟隨小編一起學(xué)習(xí)一下
2022-12-12 
Python字符串中的單詞反轉(zhuǎn)的實(shí)現(xiàn)示例
在Python中,要將字符串中的單詞進(jìn)行反轉(zhuǎn),本文主要介紹了Python字符串中的單詞反轉(zhuǎn)的實(shí)現(xiàn)示例,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(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）,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-10-10 
今天給大家?guī)淼氖顷P(guān)于Python的相關(guān)知識(shí),文章圍繞著Python pygame繪制機(jī)制展開,文中有非常詳細(xì)的介紹及圖文示例,需要的朋友可以參考下
2021-06-06