I use torch.jit.save(model.cpu(), ...) to save the cuda model to cpu torchscript. My problem is that model.cpu() changes the model from cuda to cpu, while, I need the cuda model un-changed. Is there a function like convert_to_cpu(model) that returns a cpu model without changing the model itself? Thanks.