Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.

报错信息：

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 186 and 200 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307

错误原因：

网络要求输入的featuremap大小一样，实际输入Tensor的大小不一样，我的错误中是第3位tensor有的大小是186，有的大小是200。

解决方法：

打印数据形状，把大小不一致的数据挑出剔除
对数据进行padding，在放入网络之前用np.reshape把形状统一
对于自己的图像数据，在 __getitem__ 中加上.convert('RGB')

class MyDataset(Dataset):def __init__(self, image_paths, transforms=transforms):self.image_paths = image_pathsself.transforms = transformsdef __getitem__(self, index):image = Image.open(self.image_paths[index])image = image.convert('RGB')if self.transforms:image = self.transforms(image)return image

参考：https://discuss.pytorch.org/t/runtimeerror-invalid-argument-0/17919/4

有博主写的batchsize问题，存疑，感觉不是这个问题，欢迎大家讨论

Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.相关推荐

pytorch dataload 出错：: invalid argument 0: Sizes of tensors must match except in dimension 0
错误信息:"RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0—问题分析
声明: 这里仅仅提供一个思路,暂时没有绝对正确通用的解决方案作者:{ 墨理 } 感谢通过私信.评论交流,协力探讨解决问题的同学 UP .Penser 文章目录探讨的 pytorch 报错信息如下 ...
invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1
invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 原因:dataloader 需要的数据维度是一 ...
pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
数据加载错误导致可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一样的问题,但是这里没问题我又查看了我的trans ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 ...
两种可能: 一:transforms没有把数据集中的图片转换成想要的尺寸大小,如Resnet等网络输入需要224x224大小的图像,在Resize时,不能用 transforms.Resize(224 ...
pytorch报错： invalid argument 0: Sizes of tensors must match except in dimension 0.
这个问题是在dataloader遇到的,网上一般是数据的维度不同,但我这里其实是最后一份数据和之前的数据比数量不足,所以凑不够一个batch,所以我的做法是加入drop_last=True意思是如果数 ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.
最近在修改模型不支持的算子后,由于对模型的效果产生了影响,所以需要进行微调训练,但是训练的时候报错了,如下: File "/home/xxxxxxx/EdgeNets/nn_layers/e ...
Pytorch_Geometric（PyG）使用DataLoader报错RuntimeError: Sizes of tensors must match except in dimension 0.
使用Pytorch_Geometric(PyG)时构建DataLoader,从DataLoader获取样本Batch时报错:RuntimeError: Sizes of tensors must ma ...
pytorch：data读取出错：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension
在使用Dataloader读取数据的时候,使用batch_size=1不会出现这个问题.当batch_size>1时, 默认将会使用torch.stack()为你生成一个[batch,x, x, ...

Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.

Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.相关推荐

最新文章

热门文章