search

lukecavabarrett / pna

Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric
https://arxiv.org/abs/2004.05718
MIT License
338 stars 55 forks source link

fix scheduler step #5

Closed Michaelvll closed 3 years ago

Michaelvll commented 3 years ago

Hi, I think that the mode of the scheduler should be 'max' for val_roc metric in the pytorch_geometric/example.py.

gcorso commented 3 years ago

I think you are right! Thank you for noticing!