PytorchRuntimeError: inconsistent tensor sizes at /pytorch/torch/lib/TH/generic/THTensorMath.c:2709

1.pytorch在得到dataset时没有问题,得到DataLoader时也没有问题,以batch size = 512遍历数据时,出现了:

RuntimeError: inconsistent tensor sizes at /pytorch/torch/lib/TH/generic/THTensorMath.c:2709

2.由于这个代码是我改过的版本,我用原版替代试试发现没有问题.

然后仔细对比了原版和改版的DataLoader的输出发现也没问题

之后我尝试了将batch size 改为1居然就可以跑了. 然后我去网上搜索了下发现别人在c:2864行.是由于一个批次内数据尺度不一样导致的.按batch size = 1,打印数据size,果然发现 不一致:

0 torch.Size([1, 1, 20, 36, 24]) torch.Size([1])
1 torch.Size([1, 1, 20, 36, 24]) torch.Size([1])
2 torch.Size([1, 1, 20, 36, 23]) torch.Size([1])
3 torch.Size([1, 1, 20, 36, 23]) torch.Size([1])
4 torch.Size([1, 1, 20, 24, 36]) torch.Size([1])
5 torch.Size([1, 1, 20, 24, 36]) torch.Size([1])
再回想,我在数据增广时候自作聪明的给不是立方体的数据进行了按三个面进行了90度180度270度的旋转.导致下一步裁剪时候出现了尺度不一样的情况.

3.90度180度270度的旋转要求最好是立方体(20,36,36)从(40,40,40)中取.

猜你喜欢

转载自blog.csdn.net/qq_36401512/article/details/89387599
今日推荐