PyTorch GAN RuntimeError：DataLoader工作进程中捕获了RuntimeError 0

问题描述

我的数据包含120张JPEG图像。当我想训练我的模型时，会发生错误。但是，当我使用torchvision.datasets之一而不是我的It时，效果很好。

batchSize = 64
imageSize = 64

transform=transforms.Compose([transforms.Resize(imageSize),transforms.ToTensor(),transforms.Normalize((0.5,0.5,0.5),(0.5,0.5)),])

dataset = dset.ImageFolder(root = './input/',transform = transform)
dataloader = torch.utils.data.DataLoader(dataset,batch_size = batchSize,shuffle=True,num_workers = 1)

错误：

RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "C:\Users\Szymon\Anaconda3\lib\site-packages\torch\utils\data\_utils\worker.py",line 178,in _worker_loop
    data = fetcher.fetch(index)
  File "C:\Users\Szymon\Anaconda3\lib\site-packages\torch\utils\data\_utils\fetch.py",line 47,in fetch
    return self.collate_fn(data)
  File "C:\Users\Szymon\Anaconda3\lib\site-packages\torch\utils\data\_utils\collate.py",line 79,in default_collate
    return [default_collate(samples) for samples in transposed]
  File "C:\Users\Szymon\Anaconda3\lib\site-packages\torch\utils\data\_utils\collate.py",in <listcomp>
    return [default_collate(samples) for samples in transposed]
  File "C:\Users\Szymon\Anaconda3\lib\site-packages\torch\utils\data\_utils\collate.py",line 55,in default_collate
    return torch.stack(batch,out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 64 and 65 in dimension 2 at C:\w\1\s\tmp_conda_3.7_100118\conda\conda-bld\pytorch_1579082551706\work\aten\src\TH/generic/THTensor.cpp:612
 

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）