Exploring Simple Siamese Representation Learning
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