search

awslabs / dgl-lifesci

Python package for graph neural networks in chemistry and biology
Apache License 2.0
714 stars 147 forks source link

Fix device placement in ogbg_ppa graph property prediction example #153

Closed ekorman closed 3 years ago

ekorman commented 3 years ago

Issue #, if available: The batch of graphs does not get placed to device and so fails when using GPU.

Description of changes: Set the proper device placement.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

mufeili commented 3 years ago

Thank you for the fix!