pytorch報(bào)錯(cuò)問(wèn)題

代碼調(diào)試時(shí)，顯示

ValueError: num_samples should be a positive integer value, but got num_samples=0

因?yàn)槲矣玫臄?shù)據(jù)集是已經(jīng)劃分好的，所以不需要再shuffle。

加載數(shù)據(jù)時(shí)將shuffle = False，錯(cuò)誤即可消除。

Pytorch報(bào)錯(cuò)及解決記錄

1. ValueError: num_samples should be a positive integer value, but got num_samples=0

情況描述：

一般出現(xiàn)在DataLoader(dataset=train_data, …)行。

可能原因：

傳入的dataset沒(méi)有數(shù)據(jù)，這時(shí)的 train_data.__len__() == 0，可能是函數(shù)找不到數(shù)據(jù)文件，數(shù)據(jù)路徑不對(duì)。

解決方法：

檢查自定義的 XXDataset() 類(lèi)，關(guān)于獲取數(shù)據(jù)的部分，查看路徑等是否正確。

2. RuntimeError: stack expects each tensor to be equal size, but got [1077, 3] at entry 0 and [887, 3] at entry 1

情況描述：

出現(xiàn)這個(gè)報(bào)錯(cuò)時(shí)，一般輸入的數(shù)據(jù)每個(gè)樣本的大小不相同。

可能原因：

沒(méi)有對(duì)數(shù)據(jù)進(jìn)行transforms.Resize()處理，或處理不當(dāng)，與網(wǎng)絡(luò)不匹配。

解決方法：

根據(jù)網(wǎng)絡(luò)輸入，對(duì)數(shù)據(jù)進(jìn)行transforms.Resize()處理。

3. TypeError: pic should be PIL Image or ndarray. Got <class ‘torch.Tensor’>

情況描述：

出現(xiàn)這個(gè)報(bào)錯(cuò)時(shí)，一般可能是讀取圖片數(shù)據(jù)時(shí)的格式不對(duì)，或所處理的數(shù)據(jù)不是圖片。

可能原因：

pic 數(shù)據(jù)類(lèi)型應(yīng)該是 PIL Image 或 ndarray，但卻是 torch.Tensor。

解決方法：

檢查自定義的 XXDataset() 類(lèi)，特別是 __getitem__() 方法，從讀取數(shù)據(jù)到返回，注意以下問(wèn)題：

是否有多次對(duì)數(shù)據(jù)向tensor類(lèi)型轉(zhuǎn)換，包括 torch.from_numpy()、transforms.ToTensor()等方法。

總結(jié)

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

最新評(píng)論