PyTorch Forums
Pix2pix modelling: CUDA conv_transpose2d expected weight of type CPU, got CUDA
Teymour_Aldridge
(Teymour Aldridge)
April 14, 2019, 2:15pm
1
I have now fixed the problem.