How to speed up the performance?

[SaharAlmahfouzNasser]

Hi, I am using P-GNN as a part of my network. It seems that it improves the nodes' features but my code became extremely slow!!
I checked my code and I don't think there is a bug in it. So does P-GNN slow down the performance too much?
If so what is there a reason for that?
My input graph has maximum 200 nodes.
Is there a pretrained P-GNN (Weights) that I can use?

Thank you in advance,
Sahar