Files

More Examples for Training GraphSAGE

Training with PyTorch Lightning

We provide minibatch training scripts with PyTorch Lightning in train_lightning_unsupervised.py.

Requires pytorch_lightning and torchmetrics.

python3 train_lightning_unsupervised.py