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)驗，希望能給大家一個參考，也希望大家多多支持本站。

美國服務(wù)器租用

版權(quán)聲明：本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非maisonbaluchon.cn所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場，如有內(nèi)容涉嫌侵權(quán)，請聯(lián)系alex-e#qq.com處理。