Skip to content

EdisonLeeeee/BlockGCL

Repository files navigation

BlockGCL (Under review)

This is a PyTorch implementation of BlockGCL from the paper "Oversmoothing: A Nightmare for Graph Contrastive Learning?".

Requirements

  • numpy==1.21.5
  • torch==1.12.0
  • torch-cluster==1.6.0
  • torch_geometric==2.1.0.post1
  • torch-scatter==2.0.9
  • torch-sparse==0.6.15
  • CUDA 11.6

Reproduction

To reproduce our results, please run:

bash run.sh

Due to the absence of predefined partitions for the Photo, Computer, CS, and Physics datasets, you should create a folder named "mask" in the current directory to store the random partitions.

About

PyTorch implementation of Blockwise Graph Contrastive Learning (BlockGCL)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published