To be clear, it’s awesome if this works in both cases. I just want to make sure that there is not some internal error happening which I’m unaware of and which could somehow mess up the results. I’ve written projects for work and for research in pytorch, where I used to put the model on GPU after creating the optimizer (as I was unaware of this issue) and I want to make sure I don’t need to go back and redo all the work in case something is not working.
Thank you!