What is the size of the images in your batch?  You may need to resize your images using a transform in your dataset before they can be fed into the network.  I would guess that the error is telling you that the height dimensions of some of your images in the batch are not equal.