search

lrjconan / GRAN

Efficient Graph Generation with Graph Recurrent Attention Networks, Deep Generative Model of Graphs, Graph Neural Networks, NeurIPS 2019
MIT License
469 stars 96 forks source link

Training on CPU? #2

Closed rezacsedu closed 4 years ago

rezacsedu commented 5 years ago

What changes do I have to make in order to train on CPUs, instead?

lrjconan commented 5 years ago

I have not trained it on CPU. But the code should work on CPU except scatter operator, cf. closed issue #1
You can have a simple work around to check whether it is the beginning of generation. Let me know if you encounter further issues.