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