#程式
for i, labels in enumerate(test_data):
print(12)
#報錯
Traceback (most recent call last):
File "D:\Z SOFTWARE\Anacond 3\lib\site-packages\IPython\core\interactiveshell.py", line 3325, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File "<ipython-input-12-31a3ae4ffe10>", line 1, in <module>
for i, labels in enumerate(test_data):
File "D:\Z SOFTWARE\Anacond 3\lib\torch\utils\data\dataloader.py", line 346, in __next__
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "D:\Z SOFTWARE\Anacond 3\lib\torch\utils\data\_utils\fetch.py", line 47, in fetch
return self.collate_fn(data)
File "D:\Z SOFTWARE\Anacond 3\lib\torch\utils\data\_utils\collate.py", line 79, in default_collate
return [default_collate(samples) for samples in transposed]
File "D:\Z SOFTWARE\Anacond 3\lib\torch\utils\data\_utils\collate.py", line 79, in <listcomp>
return [default_collate(samples) for samples in transposed]
File "D:\Z SOFTWARE\Anacond 3\lib\torch\utils\data\_utils\collate.py", line 55, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 215 and 258 in dimension 2 at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensor.cpp:689
a = test_data
讀取 DataLoader的資料時候報錯
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/60777.html
標籤：機器視覺
上一篇：足球？里程式猿有多遠～～
下一篇：雙倍訓直流電機調速