I am training a GAN model, i want to train it 200 epochs, but when it comes to the second epoch, an error comes:
RuntimeError: Input type (CUDAFloatTensor) and weight type (CPUFloatTensor) should be the same
I have put the data and model into gpu yet, and the first epoch training is fine, where did this error come?