Hi antspy, thanks for your help.
However, I still have something confused.
In your code, you transformed all the parameters of cuda_model.state_dict() to CPU version. What is the difference between your operation and mine? ( I did cuda_model.cpu() , I also output all the parameters of my model after executing cuda_model.cpu() and found they have all been transformed into torch.FloatTensor type).
Regards,