Okay, I got the gradients that you mentioned by rectifying the code, but still I’m unsure why the loss doesn’t converge even after a large amount of time. I’m training on a small graph of size (1000 X 1000) only. It’s not too large technically when the model needs to be tested and evaluated on graphs in the size of a million nodes!