Could you try to concatenate both images in crop_half?
up = transforms...
down = transforms...
return torch.cat((up, down), dim=2)

PS: If you post code using three backticks ```, we could easily copy your code and debug it.
Also the search will be able to use your code so that others could easily find your solution to a similar problem.