After moving a model to CPU to execute a step, I would like to move it to GPU for the next steps. But when I do this, the weight remains on CPU. I am using .to() method. I see the error ‘RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same’
What might be going on?