ptrblck:

You cannot move the computation graph to the CPU after its creation by moving leaf variables to the CPU (at least I’m not aware of a method to do so).


Alright, that’s not what I have hoped for  but thank you a lot for your help!