@887574002 I am trusting you verified the elements of the batch to be images. But, just in case, is it possible that the batch returns a tuple (of size 2) where each element is a tensor of shape batch_size x shape_of_transformed_image?