Solved by ptrblck in post #4 
              
              
                Sorry for forgetting the second part of the right approach.  
The new error is not raised by input.clone(), but should still be raised by self.latest_input, which is now on the CPU as it wasn’t properly registered. 
To make sure model.to() pushes all internal parameters and buffers to the desired de…