欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

Pytorch測試神經(jīng)網(wǎng)絡時出現(xiàn) RuntimeError:的解決方案

 更新時間：2021年05月28日 09:58:44   作者：漂洋過海的油條  
這篇文章主要介紹了Pytorch測試神經(jīng)網(wǎng)絡時出現(xiàn) RuntimeError:的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

Pytorch測試神經(jīng)網(wǎng)絡時出現(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')

補充：解決RuntimeError: Error(s) in loading state_dict for XXXX

在運行代碼時遇到了這個錯誤，顯示錯誤對應代碼中的state_dict，找到對應的語句

修改前：

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 可以用來判斷模型當前運行環(huán)境與之前是否相同，在默認情況下是True（純屬個人理解，勿噴）

補充：使用Pytorch訓練模型出現(xiàn)RuntimeError: CUDA out of memory錯誤解決

訓練：

由于GPU顯存資源有限，訓練輸入的batchsize不能過大，過大會導致out of memory錯誤。

解決方案：

將batchsize減小，甚至是為1

測試時出現(xiàn)此問題解決方案：

在測試代碼之前使用 with torch.no_grad():

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

相關(guān)文章

  • pytorch geometric的GNN、GCN的節(jié)點分類方式

    pytorch geometric的GNN、GCN的節(jié)點分類方式

    這篇文章主要介紹了pytorch geometric的GNN、GCN的節(jié)點分類方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
    2022-12-12
  • Python根據(jù)字典值對字典進行排序的三種方法實例

    Python根據(jù)字典值對字典進行排序的三種方法實例

    Python中的字典是無序類型,沒有自己的排序方法,下面這篇文章主要給大家介紹了關(guān)于Python根據(jù)字典值對字典進行排序的三種方法,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下
    2022-09-09
  • Python基于matplotlib畫箱體圖檢驗異常值操作示例【附xls數(shù)據(jù)文件下載】

    Python基于matplotlib畫箱體圖檢驗異常值操作示例【附xls數(shù)據(jù)文件下載】

    這篇文章主要介紹了Python基于matplotlib畫箱體圖檢驗異常值操作,涉及Python針對xls格式數(shù)據(jù)文件的讀取、matplotlib圖形繪制等相關(guān)操作技巧,并附帶xls數(shù)據(jù)文件供讀者下載參考,需要的朋友可以參考下
    2019-01-01
  • Python Opencv中用compareHist函數(shù)進行直方圖比較對比圖片

    Python Opencv中用compareHist函數(shù)進行直方圖比較對比圖片

    這篇文章主要介紹了Python Opencv中用compareHist函數(shù)進行直方圖比較進行對比圖片，本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下
    2020-04-04
  • python中如何設置list步長

    python中如何設置list步長

    這篇文章主要介紹了python中如何設置list步長，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
    2022-05-05
  • Python使用pandas將表格數(shù)據(jù)進行處理

    Python使用pandas將表格數(shù)據(jù)進行處理

    這篇文章主要介紹了Python使用pandas將表格數(shù)據(jù)進行處理，文章圍繞主題展開詳細的內(nèi)容介紹，具有一定的參考價值，感興趣的小伙伴可以參考一下
    2022-08-08
  • 利用 Python 實現(xiàn)多任務進程

    利用 Python 實現(xiàn)多任務進程

    這篇文章主要介紹如何利用 Python 實現(xiàn)多任務進程，正在執(zhí)行的程序，由程序、數(shù)據(jù)和進程控制塊組成，是正在執(zhí)行的程序，程序的一次執(zhí)行過程，是資源調(diào)度的基本單位。下面就來詳細介紹改內(nèi)容,需要的朋友可以參考一下
    2021-10-10
  • Python數(shù)據(jù)分析:手把手教你用Pandas生成可視化圖表的教程

    Python數(shù)據(jù)分析:手把手教你用Pandas生成可視化圖表的教程

    今天小編就為大家分享一篇Python數(shù)據(jù)分析:手把手教你用Pandas生成可視化圖表的教程，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-12-12
  • python3.6使用urllib完成下載的實例

    python3.6使用urllib完成下載的實例

    今天小編就為大家分享一篇python3.6使用urllib完成下載的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-12-12
  • python中的變量命名規(guī)則詳情

    python中的變量命名規(guī)則詳情

    這篇文章主要介紹了python中的變量命名規(guī)則詳情，變量名可以包括字母、數(shù)字、下劃線，但是數(shù)字不能做為開頭，變量用的好或不好，和代碼質(zhì)量有著非常重要的聯(lián)系，合理的使用變量，可以讓你的代碼可讀性更高并且更加簡潔,下面相關(guān)內(nèi)容吧需要的小伙伴可以參考一下
    2022-03-03

最新評論