dmlc--dgl
Chang Liu
13 行
306 B
Markdown
13 行
306 B
Markdown
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`.
|
|
|
|
```bash
|
|
python3 train_lightning_unsupervised.py
|
|
```
|