To be more specific, after I have run for 100 iteration, I attempted to save my model. And then it works…But it somehow screw up my gpu setting in the program and make the model and input not no the same device anymore…