how to check my model is on the gpu in pytorch code example Example: how to check weather my model is on gpu in pytorch next(network.parameters()).device