Can I see the code where you give the model the input. This error is happening because either the input or the model has not been moved to cuda. So you need to move either the input or model to cuda whichever you haven’t done yet.