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

Fixed memory leak #14

Open euanong opened 2 years ago

euanong commented 2 years ago

Disabled autograd during testing to avoid OOM issues on large graphs for machines with limited memory.