Pytorch測(cè)試神經(jīng)網(wǎng)絡(luò)時(shí)出現(xiàn) RuntimeError:的解決方案
Pytorch測(cè)試神經(jīng)網(wǎng)絡(luò)時(shí)出現(xiàn)“RuntimeError: Error(s) in loading state_dict for Net”
解決方法：
load_state_dict(torch.load('net.pth')
在前，增加
model = nn.DataParallel(model)
就可以了。
比如
net = NET()
net.cuda()
net = nn.DataParallel(net)
net.load_state_dict(torch.load('net.pth')
補(bǔ)充：解決RuntimeError: Error(s) in loading state_dict for XXXX
在運(yùn)行代碼時(shí)遇到了這個(gè)錯(cuò)誤，顯示錯(cuò)誤對(duì)應(yīng)代碼中的state_dict，找到對(duì)應(yīng)的語(yǔ)句
修改前：
net.load_state_dict(torch.load(model_para_path))
修改后：
net.load_state_dict(torch.load(model_para_path),False)
我遇到的問(wèn)題得到解決。經(jīng)查閱資料，我理解的原因是state_dict的四個(gè)參數(shù)之一:_module 可以用來(lái)判斷模型當(dāng)前運(yùn)行環(huán)境與之前是否相同，在默認(rèn)情況下是True（純屬個(gè)人理解，勿噴）
補(bǔ)充：使用Pytorch訓(xùn)練模型出現(xiàn)RuntimeError: CUDA out of memory錯(cuò)誤解決
訓(xùn)練：
由于GPU顯存資源有限，訓(xùn)練輸入的batchsize不能過(guò)大，過(guò)大會(huì)導(dǎo)致out of memory錯(cuò)誤。
解決方案：
將batchsize減小，甚至是為1
測(cè)試時(shí)出現(xiàn)此問(wèn)題解決方案：
在測(cè)試代碼之前使用 with torch.no_grad():
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
- Pytorch 神經(jīng)網(wǎng)絡(luò)—自定義數(shù)據(jù)集上實(shí)現(xiàn)教程
- 使用pytorch提取卷積神經(jīng)網(wǎng)絡(luò)的特征圖可視化
- Python深度學(xué)習(xí)pytorch神經(jīng)網(wǎng)絡(luò)Dropout應(yīng)用詳解解
- Python深度學(xué)習(xí)pytorch神經(jīng)網(wǎng)絡(luò)多輸入多輸出通道
- Python深度學(xué)習(xí)pytorch實(shí)現(xiàn)圖像分類數(shù)據(jù)集
- 總結(jié)近幾年P(guān)ytorch基于Imgagenet數(shù)據(jù)集圖像分類模型
- Pytorch搭建簡(jiǎn)單的卷積神經(jīng)網(wǎng)絡(luò)(CNN)實(shí)現(xiàn)MNIST數(shù)據(jù)集分類任務(wù)
相關(guān)文章
Python小紅書(shū)旋轉(zhuǎn)驗(yàn)證碼識(shí)別實(shí)戰(zhàn)教程
這篇文章主要介紹了Python小紅書(shū)旋轉(zhuǎn)驗(yàn)證碼識(shí)別實(shí)戰(zhàn)教程,本文通過(guò)示例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友參考下吧2023-08-08
Python和Java進(jìn)行DES加密和解密的實(shí)例
下面小編就為大家分享一篇Python和Java進(jìn)行DES加密和解密的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-01-01
Python爬蟲(chóng)入門教程01之爬取豆瓣Top電影
這篇文章主要介紹了Python爬蟲(chóng)入門教程01：豆瓣Top電影爬取的方法，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2021-01-01
Task主要作用是幫助event_loop調(diào)度,如果Task發(fā)現(xiàn)它打包的協(xié)程在await一個(gè)對(duì)象(執(zhí)行一個(gè)io操作),則Task會(huì)掛起該協(xié)程的執(zhí)行,把控制權(quán)交回event_loop,讓他去運(yùn)行其他Task,當(dāng)該對(duì)象運(yùn)行完畢,Task將打包的協(xié)程恢復(fù)運(yùn)行,本文介紹Python asyncio使用小結(jié),感興趣的朋友一起看看吧2024-02-02
在本篇文章里小編給大家整理了一篇關(guān)于python匿名函數(shù)的實(shí)例用法，有興趣的朋友們可以學(xué)習(xí)參考下。2021-03-03
django生產(chǎn)環(huán)境搭建(uWSGI+django+nginx+python+MySQL)
本文主要介紹了django生產(chǎn)環(huán)境搭建，主要包括uWSGI+django+nginx+python+MySQL，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-08-08
TorchVision Transforms API目標(biāo)檢測(cè)實(shí)例語(yǔ)義分割視頻類
這篇文章主要為大家介紹了TorchVision Transforms API大升級(jí)，支持目標(biāo)檢測(cè)、實(shí)例/語(yǔ)義分割及視頻類任務(wù)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-11-11
Python+Pyqt實(shí)現(xiàn)簡(jiǎn)單GUI電子時(shí)鐘
這篇文章主要為大家詳細(xì)介紹了Python+Pyqt實(shí)現(xiàn)簡(jiǎn)單GUI電子時(shí)鐘，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-07-07
OpenCV HSV顏色識(shí)別及HSV基本顏色分量范圍
這篇文章主要介紹了OpenCV HSV顏色識(shí)別及HSV基本顏色分量范圍，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-03-03
淺談keras中的Merge層(實(shí)現(xiàn)層的相加、相減、相乘實(shí)例)
這篇文章主要介紹了淺談keras中的Merge層(實(shí)現(xiàn)層的相加、相減、相乘實(shí)例)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-05-05