Hi, thanks for the inspiring work of GCC.

I wonder if there is a way to run GCC pretrain on CPU-only. As it appears to me, there is no easy way (such as specifying an option in the arguments) to do this.

P.S. Installation of RDKit is really painful for servers inside mainland china. Through checking the code, I found that there is no need to install RDKit, all one have to do is to copy the GAT and GCN layer code of DGL to the models/gat.py and models/gcn.py respectively. Please kindly correct me if I was wrong.

Any help is highly appreciated : )