How is the performance of GraphNeuralNetworks.jl compared to PytorchGeometric?

DoktorMike
Wed, 12 Apr 2023 15:27:13 +0000
Machine Learning

I’ve been playing around with the excellent GraphNeuralNetworks.jl and am thinking about applying it to the biochemistry domain. Specifically within drug discovery. I was wondering if anyone has tried to benchmark speed on CPU and/or GPU for training the same network as compared to Pytorch Geometric?

Also if there are any gotchas that I should be on the lookout for? The tasks I’m thinking about are both node level predictions but also full graph predictions.

Sorry for the vague question.

//Mike

5 posts - 2 participants