Self-supervised representation learning with tf-gnn

[novaczkisz]

Hi TF-GNN users!

Could anyone share example code demonstrating the use of contrastive losses/DGI module, or suggest an alternative method to set up a basic self-supervised representation learning task? Additionally, if you could point me to a repository that utilizes tf-gnn in this context, that would also be great.

Thanks,
Szabolcs