dmlc--dgl
704bcaf6dd
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
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