mirrors / dmlc--dgl

文件历史

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

3 次代码提交

作者 SHA1 备注 提交日期
Tingyu Wang bcf9923b32 [Model] Add dgl.nn.CuGraphSAGEConv model (#5137) 
* add CuGraphSAGEConv model

* fix lint issues

* update model to reflect changes in make_mfg_csr(), move max_in_degree to forward()

* lintrunner

* allow reset_parameters()

* remove norm option, simplify test

* allow full graph fallback option, add example

* address comments

* address reviews

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-22 13:43:28 +08:00
Tingyu Wang 19b3cea166 [Model] Update CuGraphRelGraphConv to use pylibcugraphops=23.02 (#5217) 
* update cugraph_relgraphconv

* update equality test

* update cugraph rgcn example

* update RelGraphConvAgg based on latest API changes

* enable fallback option to fg when fanout is large

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-15 20:11:20 +08:00
Tingyu Wang 943abba3fb [Example] add entity_sample example of using CuGraphRelGraphConv (#4827) 
* add entity_sample example

* update message

* address review, update api call

* Update examples/pytorch/rgcn/entity_sample_cugraph_relgraphconv.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update examples/pytorch/rgcn/entity_sample_cugraph_relgraphconv.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update examples/pytorch/rgcn/entity_sample_cugraph_relgraphconv.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* address review

* move example and add description

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-12-01 21:09:14 +08:00