Solved by ptrblck in post #2 
              
              
                Keras might transfer the model parameters and data automatically for you behind the scenes, while you are responsible to do it in PyTorch. 
I guess it comes down to balance “convenience vs. flexibility” and I personally like to manually specify which tensor is places in which device, which e.g. enab…