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)文章
Python中shapefile轉(zhuǎn)換geojson的示例
今天小編就為大家分享一篇關(guān)于Python中shapefile轉(zhuǎn)換geojson的示例，小編覺得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來看看吧2019-01-01
今天小編就為大家分享一篇基于Python中的yield表達(dá)式詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-11-11
python機(jī)器基礎(chǔ)邏輯回歸與非監(jiān)督學(xué)習(xí)
這篇文章主要為大家介紹了python機(jī)器基礎(chǔ)邏輯回歸與非監(jiān)督的學(xué)習(xí)講解u，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步2021-11-11
這篇文章主要介紹了Python?__all__變量用法,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2023-04-04
這篇文章主要介紹了python中如何以空格為分割符,給列表賦予數(shù)值問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
這篇文章主要介紹了Spy++的使用方法及下載教程,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-01-01
Python使用Pickle模塊進(jìn)行數(shù)據(jù)保存和讀取的講解
今天小編就為大家分享一篇關(guān)于Python使用Pickle模塊進(jìn)行數(shù)據(jù)保存和讀取的講解，小編覺得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來看看吧2019-04-04
利用Python批量壓縮png方法實(shí)例（支持過濾個(gè)別文件與文件夾）
這篇文章主要給大家介紹了關(guān)于利用Python批量壓縮png的相關(guān)資料，文中介紹的方法支持過濾個(gè)別文件與文件夾，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友們下面跟著小編來一起看看吧。2017-07-07
PyMySQL?遵循?Python?數(shù)據(jù)庫?API?v2.0?規(guī)范，并包含了?pure-Python?MySQL?客戶端庫，這篇文章主要介紹了Spring?DI依賴注入詳解,需要的朋友可以參考下2022-11-11
python批量處理多DNS多域名的nslookup解析實(shí)現(xiàn)
這篇文章主要介紹了python批量處理多DNS多域名的nslookup解析實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06