Calling methods after moving model to gpu

When I move a model to a gpu using to(device), am I still able to call methods within the model itself (like within forward) that aren’t gpu compatible? For example, I’d like to process a graph Data object in the forward method but doing so would rely on packages not meant for the gpu.

Yes, this would be possible, but depending on the code logic your code might be synchronizing at this point. e.g. if the result of this CPU operation is needed in the forward for some operations.