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 |
|