RuntimeError: stack expects each tensor to be equal size, but got [3, 500, 656] at entry 0 and [3, 5

RuntimeError: stack expects each tensor to be equal size, but got [3, 500, 656] at entry 0 and [3, 500, 666] at entry 1

这个问题出现的原因是因为输入dataloader的图片大小并不一致。注意检查train, test, val数据集是否都进行了图片大小处理。如果没有的话，可以使用

 transforms.Resize((256, 256))

改变图片大小也可以使用RandomResizedCrop， Scale, Centercrop等。