Pix2pix modelling: CUDA conv_transpose2d expected weight of type CPU, got CUDA

I have now fixed the problem.