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