解決pytorch 數(shù)據(jù)類(lèi)型報(bào)錯(cuò)的問(wèn)題
pytorch報(bào)錯(cuò):
RuntimeError: Expected object of type Variable[torch.LongTensor] but found type Variable[torch.cuda.ByteTensor] for argument #1 ‘a(chǎn)rgument1'
解決方法:
pytorch框架在存儲(chǔ)labels時(shí),采用LongTensor來(lái)存儲(chǔ),所以在一開(kāi)始dataset返回label時(shí),就要返回與LongTensor對(duì)應(yīng)的數(shù)據(jù)類(lèi)型,即numpy.int64
補(bǔ)充:使用pytorch遇到的各種問(wèn)題及解決方案
自己在使用pytorch遇到的各種問(wèn)題及解決方案:
RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'
RuntimeError: The size of tensor a (12800) must match the size of tensor b (100) at non-singleton dimension 0
輸入維度為12800,輸出維度為100,輸入輸出的維度不一致,正確的例子如下:
inputs = [(1,2,3), (2,3,4)] outsputs = [4, 5]
將輸入輸出的長(zhǎng)度改為一致
取tensor的第一個(gè)元素
XXX.item() # XXX為tensor對(duì)象
tensor中的元素改變數(shù)據(jù)類(lèi)型
# 常常因?yàn)閿?shù)據(jù)類(lèi)型出錯(cuò),要修改數(shù)據(jù)類(lèi)型 XXX.int() XXX.float()
補(bǔ)充:Pytorch的Dataloader報(bào)錯(cuò):TypeError: batch must contain tensors, numbers, dicts or lists
具體報(bào)錯(cuò):
TypeError: batch must contain tensors, numbers, dicts or lists; found <class 'PIL.Image.Image'>
loader的代碼:
dataloader=torch.utils.data.DataLoader(dataset,batch_size=1,shuffle=True)
表面上看這個(gè)代碼沒(méi)有問(wèn)題,實(shí)際上問(wèn)題出在了dataloader機(jī)制的要求上,dataloader要求接收的是一個(gè)tensor,而我的dataset沒(méi)有做transform,所以dataset的getitem函數(shù)返回的是一個(gè)PIL的Image對(duì)象,所以就會(huì)報(bào)錯(cuò)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
相關(guān)文章
解決Python pandas df 寫(xiě)入excel 出現(xiàn)的問(wèn)題
今天小編就為大家分享一篇解決Python pandas df 寫(xiě)入excel 出現(xiàn)的問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07
python實(shí)現(xiàn)圖書(shū)館搶座(自動(dòng)預(yù)約)功能的示例代碼
這篇文章主要介紹了python實(shí)現(xiàn)圖書(shū)館搶座(自動(dòng)預(yù)約)功能,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-09-09
Python運(yùn)行報(bào)錯(cuò)UnicodeDecodeError的解決方法
本文給大家分享的是在Python項(xiàng)目中經(jīng)常遇到的關(guān)于編碼問(wèn)題的一個(gè)小bug的解決方法以及分析方法,有相同遭遇的小伙伴可以來(lái)參考下2016-06-06
Python使用gluon/mxnet模塊實(shí)現(xiàn)的mnist手寫(xiě)數(shù)字識(shí)別功能完整示例
這篇文章主要介紹了Python使用gluon/mxnet模塊實(shí)現(xiàn)的mnist手寫(xiě)數(shù)字識(shí)別功能,結(jié)合完整實(shí)例形式分析了Python調(diào)用gluon/mxnet模塊識(shí)別手寫(xiě)字的具體實(shí)現(xiàn)技巧,需要的朋友可以參考下2019-12-12
python實(shí)現(xiàn)柵欄加解密 支持密鑰加密
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)柵欄加解密,支持密鑰加密,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-03-03
PyCharm 配置遠(yuǎn)程python解釋器和在本地修改服務(wù)器代碼
這篇文章主要介紹了PyCharm 配置遠(yuǎn)程python解釋器和在本地修改服務(wù)器代碼,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
python上selenium的彈框操作實(shí)現(xiàn)
這篇文章主要介紹了python上selenium的彈框操作實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07

