In rexgen example: Enable training on GPU and evaluation on CPU

awslabs / dgl-lifesci

Python package for graph neural networks in chemistry and biology

Apache License 2.0

714 stars 147 forks source link

Closed mar-volk closed 3 years ago

mar-volk commented 3 years ago

Issue #130

Description of changes: In rexgen example: Enable training on CUDA-supporting maching and evaluation on non-CUDA-supporting machine.

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