search

Shen-Lab / GraphCL

[NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen
MIT License
548 stars 103 forks source link

About Unsupervised_Cora_Citeseer #51

Open xyf2020 opened 2 years ago

xyf2020 commented 2 years ago

In the file execute.py , perhaps you should state the CUDA environment before import torch package. Otherwise, only the default GPU could be used.

yyou1996 commented 2 years ago

Hi @xyf2020,

Thanks for the kind reminder! We will try to consummate the document as much as we can. 💪