Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Feature] Unify self-supervised training and add function to save results #135

Merged
merged 27 commits into from
Dec 31, 2020
Merged

[Feature] Unify self-supervised training and add function to save results #135

merged 27 commits into from
Dec 31, 2020

Conversation

THINK2TRY
Copy link
Contributor

Description

  1. Unify the training mode of self-supervised model (DGI/MVGRL/Unsupervised GraphSAGE/GRACE) with SelfSupvervisesTrainer
  2. Add the function to save model checkpoints and embeddings.

Checklist

Please feel free to remove inapplicable items for your PR.

  • The PR title starts with [$CATEGOTY] (such as [Model], [Doc], [Feature], [Bugfix])
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • To my best knowledge, examples are either not affected by this change,
    or have been fixed to be compatible with this change

@THINK2TRY
Add graph unet
fe0852a
@THINK2TRY
Add dropout-adj and dropout for u-net
789e9e4
@THINK2TRY
Fix bugs in hyper-parameters search examples
0811c6c
@THINK2TRY
Add PPRGo for large scale graph training
a9559ce
@THINK2TRY
Add test and example for PPRGo
0d403b1
@THINK2TRY
Merge branch 'master' into pprgo
194fb7c
@THINK2TRY
Replace set_grad_enabled with no_grad in PPRGoTrainer
2f2f103
@THINK2TRY
Merge branch 'pprgo' of https://github.com/THINK2TRY/cogdl into pprgo
2e0a022
@THINK2TRY
Fix bugs in examples/hyper_search
4d51720
@THINK2TRY
Modify pytorch version from 1.5 to 1.6 in travis
f6a99f0
@THINK2TRY
Fix a bug in ppr_trainere
5f680a1
@THINK2TRY
Merge upstream/master
bc98bd8
@THINK2TRY
Fix black
3868366
@THINK2TRY
Merge branch 'master' into unet
32a7f54
@THINK2TRY
Add test for ProNE_module and modify ppr_module
f7ef798
@THINK2TRY
Merge branch 'unet' of https://github.com/THINK2TRY/cogdl into unet
ba430e3
@THINK2TRY
Merge upstream
1d74c12
@THINK2TRY
Merge remote-tracking branch 'origin/unet'
877153a
@THINK2TRY
Merge branch 'master' into pprgo
5594bcd
@THINK2TRY
Modify DGI/MVGRL/GraphSAGE for self_supervised_trainer
c405f13
@THINK2TRY
Modify self_supervised_trainer to unify training paradigm
c864bc0
@THINK2TRY
Save model checkpoint and unsupervised representation embeddings
4ffb4bc
@THINK2TRY
Fix bugs
52547ae
@THINK2TRY
Merge remote-tracking branch 'upstream/master'
87540d5
@THINK2TRY
Merge branch 'master' into save_checkpoint
fc3a9f7
@THINK2TRY
Ignore saving model for link prediction and fix bugs in unsup-node-cl…
d6d6da7 
…assfication test
@THINK2TRY
Fix bugs in link prediction test
5e4bc80
@THINK2TRY THINK2TRY merged commit dcf5493 into THUDM:master Dec 31, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@THINK2TRY