Graph GPS example on Zinc

[arijitthegame]

Running the example code : https://github.com/pyg-team/pytorch_geometric/blob/master/examples/graph_gps.py

gives extremely poor results. Can someone please let me know what should be the right hyperparameters? I am guessing we should be expecting something around .226. Also the number of parameters for the model is 423,857 whereas this paper (https://arxiv.org/pdf/2106.05234 table 4) reports 588,929 parameters. Thank you.