3 Star 1 Fork 0

Gitee 极速下载/GraphLearn-for-PyTorch

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
此仓库是为了提升国内下载速度的镜像仓库,每日同步一次。 原始仓库: https://github.com/alibaba/graphlearn-for-pytorch
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

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
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mirrors/GraphLearn-for-PyTorch.git
git@gitee.com:mirrors/GraphLearn-for-PyTorch.git
mirrors
GraphLearn-for-PyTorch
GraphLearn-for-PyTorch
main

搜索帮助