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.