Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
README.md 306 Bytes
Newer Older
Quan (Andy) Gan's avatar
[Doc and bugfix] Add docs and user guide and update tutorial for sampling pipeline (#3774)  
Quan (Andy) Gan committed
1
2
3
4
5 
More Examples for Training GraphSAGE
============================

### Training with PyTorch Lightning
Chang Liu's avatar
[Example][Refactor] Multi-gpu graphsage link prediction example refactor (#5091)  
Chang Liu committed
6 
We provide minibatch training scripts with PyTorch Lightning in `train_lightning_unsupervised.py`.
Quan (Andy) Gan's avatar
[Doc and bugfix] Add docs and user guide and update tutorial for sampling pipeline (#3774)  
Quan (Andy) Gan committed
7
8
9
10
11
12 

Requires `pytorch_lightning` and `torchmetrics`.

```bash
python3 train_lightning_unsupervised.py
```