Hi @yyou1996, Thanks for your efforts in this project. The bug has been fixed, which is due to the pyG's inconsistent with the specified version (1.0.3). The author has mentioned it [here](https://github.com/snap-stanford/pretrain-gnns/issues/14).