Hi,
It looks like all images in your dataset are not the same size: the loader fails to concatenate them because some have size 32 on the 3rd dimension and some have size 48 on this same dimension.
Are you sure that your dataset contains images of size 128*128? make sure that the getitem function returns a tensor of size number_channels x 128 x 128.