Bug report for transfer learning.

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

547 stars 103 forks source link

Closed ha-lins closed 3 years ago

ha-lins commented 3 years ago

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.

huangzizheng01 commented 1 year ago

Two things for this bug in transferLearning_MoleculeNet_PPI\bio may help (maybe similar in chem):

replace cat_dim to __cat_dim__ in loader.py line177
replacereturn self.propagate(self.aggr, edge_index, x=x, edge_attr=edge_embeddings) in model.py line52 with return self.propagate(edge_index[0], x=x, edge_attr=edge_embeddings) and replace line111 with return self.propagate(edge_index[0], x=x, edge_attr=edge_embeddings, norm = norm)