search

Shen-Lab / GraphCL_Automated

[ICML 2021] "Graph Contrastive Learning Automated" by Yuning You, Tianlong Chen, Yang Shen, Zhangyang Wang; [WSDM 2022] "Bringing Your Own View: Graph Contrastive Learning without Prefabricated Data Augmentations" by Yuning You, Tianlong Chen, Zhangyang Wang, Yang Shen
109 stars 11 forks source link

About pretraining models #6

Open StylesZhang opened 2 years ago

StylesZhang commented 2 years ago

Hi! Could you please release your saved model file of transfer learning in JOAO? I mean the weights file that will be used in finetuning of transfer learning.

yyou1996 commented 2 years ago

Hi @HeyMercer,

Thank you for your comments. Checkpoints can be very easily generated following the scripts https://github.com/Shen-Lab/GraphCL_Automated/tree/master/transferLearning_MoleculeNet_PPI#joao-pre-training, with an estimated time < 2 days. Feel free to pin me if something doesn't work.