- Multi-processing Feature usage.
python feature_mp.py
- Single GPU basic GraphSAGE on OGBN-Products similar to PyG.
This example is implemeted using
NeighborSampler
.
python train_sage_ogbn_products.py
- Multiple GPUs GraphSAGE on OGBN-papers100M.
python multi_gpu/train_sage_ogbn_papers100m.py
- Heterogeneous GraphSage examples on OGBN-MAG compatible with PyG.
# single GPU
python hetero/train_hgt_mag.py
# multi-GPUs
python hetero/train_hgt_mag_mp.py
- Training on PAI.
see pai/README.md
- Distributed (multi nodes) examples. see distributed/README.md