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

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

test_transform = transforms.Compose([
    transforms.Resize((256, 256)),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225])
])

重点是 transforms.Resize((256, 256))这一行。除此以外，改变图片大小也可以使用RandomResizedCrop， Scale, Centercrop等。