I'm training a Graph Neural Network (GNN) model for link prediction in PyTorch. I’ve taken several