I am trying to create a Geometric Deep Learning model using Pytorch. I have around 5000 graphs, which are split into a training set, validation set and test set. Every graph has one ‘correct’ node and is fully connected. There are different node and edge features used. Before loading the graphs into the model, they are split into batches.