The error is telling you that you are trying to put images of different size (size 1: [3, 900, 1600], size 2: [3, 1500, 2000]) into one batch. This is not possible tho. The images need to be of the same size.