Skip to content

Files

Latest commit

0e566dc · Jun 13, 2024

History

History
This branch is 11 commits behind alibaba/graphlearn-for-pytorch:main.

examples

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Apr 26, 2024
Jan 8, 2024
Jun 21, 2023
Apr 29, 2024
Mar 28, 2023
Jun 13, 2024
Mar 28, 2023
Jun 21, 2023
Apr 28, 2023
Jun 21, 2023
Mar 28, 2023
Jun 1, 2023

GNN examples

  1. Multi-processing Feature usage.
python feature_mp.py
  1. Single GPU basic GraphSAGE on OGBN-Products similar to PyG. This example is implemeted using NeighborSampler.
python train_sage_ogbn_products.py
  1. Multiple GPUs GraphSAGE on OGBN-papers100M.
python multi_gpu/train_sage_ogbn_papers100m.py
  1. 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
  1. Training on PAI.

see pai/README.md

  1. Distributed (multi nodes) examples. see distributed/README.md