G-SimSiam

A PyTorch implementation which refers to repo for the paper Exploring Simple Siamese Representation Learning by Xinlei Chen & Kaiming He

Additional functionals

  • support graph data
  • additional graph augmentations is added

Training

  • training gcn on ciaw dataset
    CUDA_VISIBLE_DEVICES=0 python train_ciaw.py –data_dir ../Data/ –log_dir ./logs/ -c configs/simsiam_ciaw.yaml –ckpt_dir ~/.cache/ –hide_progress

GitHub

View Github