mirrors / dmlc--dgl

文件历史

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

17 次代码提交

作者 SHA1 备注 提交日期
Chang Liu 533afa851e [Example][Refactor] Graphsage link prediction example refactor (#4526) 
* Refactor link pred example for graphsage

* Use ogb evaluator + README update

* Update

* Add comments
 2022-09-23 16:45:32 +08:00
peizhou001 d077d3719c [EXAMPLE]Add multi gpu graph predication GIN+virtualnode example (#4385) 
* add multigpu folder for related examples
 2022-08-19 09:38:19 +08:00
Chang Liu b76d0ed1db [Example][Refactor] Regolden graphsage example for future guide (#4186) 
* Regolden graphsage example to guide others

* update golden

* update

* Update example and propagate to original folder

* Update to remove ^M (windows DOS) character

* update

* Merge file changes and update README

* Minor comment update

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-06-30 21:56:13 +08:00
Quan (Andy) Gan 0d878ff8ef [Example] Cleaned GraphSAGE node classification example with PyTorch Lightning (#3863) 
* cleaned pl node classification example

* conform to PL's method of updating the dataloader

* update

* lint

* fix test

* fix
 2022-04-12 17:26:23 +08:00
Mufei Li 01e50626d3 [Example] Add TorchMetrics in README (#3913) 
* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2022-04-09 23:25:51 +08:00
nv-dlasalle 57d2f31f20 [Examples] Add pure gpu example of graphsage (#3796) 
* Add pure_gpu example of graphsage

* move to advanced directory

Co-authored-by: Quan Gan <coin2028@hotmail.com>
 2022-03-10 17:33:23 -08:00
Quan (Andy) Gan d41d07d0f6 [Doc and bugfix] Add docs and user guide and update tutorial for sampling pipeline (#3774) 
* huuuuge update

* remove

* lint

* lint

* fix

* what happened to nccl

* update multi-gpu unsupervised graphsage example

* replace most of the dgl.mp.process with torch.mp.spawn

* update if condition for use_uva case

* update user guide

* address comments

* incorporating suggestions from @jermainewang

* oops

* fix tutorial to pass CI

* oops

* fix again

Co-authored-by: Xin Yao <xiny@nvidia.com>
 2022-02-28 02:21:51 +08:00
Xin Yao 738e8318fd [Feature] CUDA UVA sampling for MultiLayerNeighborSampler (#3674) 
* implement pin_memory/unpin_memory/is_pinned for dgl.graph

* update python docstring

* update c++ docstring

* add test

* fix the broken UnifiedTensor

* XPU_SWITCH for kDLCPUPinned

* a rough version ready for testing

* eliminate extra context parameter for pin/unpin

* update train_sampling

* fix linting

* fix typo

* multi-gpu uva sampling case

* disable new format materialization for pinned graphs

* update python doc for pin_memory_

* fix unit test

* UVA sampling for link prediction

* dispatch most csr ops

* update graphsage example to combine uva sampling and UnifiedTensor

* update graphsage example to combine uva sampling and UnifiedTensor

* update graphsage example to combine uva sampling and UnifiedTensor

* update doc

* update examples

* change unitgraph and heterograph's PinMemory to in-place

* update examples for multi-gpu uva sampling

* update doc

* fix linting

* fix cpu build

* fix is_pinned for DistGraph

* fix is_pinned for DistGraph

* update graphsage unsupervised example

* update doc for gpu sampling

* update some check for sampling device switching

* fix linting

* adapt for new dataloader

* fix linting

* fix

* fix some name issue

* adjust device check

* add unit test for uva sampling & fix some zero_copy bug

* fix linting

* update num_threads in graphsage examples

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
 2022-02-09 14:06:17 +08:00
Mufei Li 3f6f694159 [Bug fix] Various fix from bug bash (#3133) 
* Update

* Update

* Update dependencies

* Update

* Update

* Fix ogbn-products gat

* Update

* Update

* Reformat

* Fix typo in node2vec_random_walk

* Specify file encoding

* Working for 6.7

* Update

* Fix subgraph

* Fix doc for sample_neighbors_biased

* Fix hyperlink

* Add example for udf cross reducer

* Fix

* Add example for slice_batch

* Replace dgl.bipartite

* Fix GATConv

* Fix math rendering

* Fix doc

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-17.us-west-2.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-22-156.us-west-2.compute.internal>
 2021-07-15 09:17:15 +08:00
Quan (Andy) Gan 70695ff8f8 [Model] Training GraphSAGE with PyTorch Lightning (#2878) 
* pytorch lightning initial examples

* revert most changes in dataloader to favor #2886.

* address comments
 2021-05-11 11:28:13 +08:00
Quan (Andy) Gan 4ddd477f35 [Model] GraphSAGE inductive example (#1741) 
* graphsage inductive example

* fix
 2020-07-03 11:32:06 +08:00
Quan (Andy) Gan dc0432e737 [Model] Unsupervised learning with GraphSAGE (#1440) 
* unsupervised graphsage first commit

* fix

* disable remove_edges and still got 0.90 performance

* optimize edgeids with multimap

* change hyperparams

* update README

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2020-04-28 17:19:33 +08:00
Quan (Andy) Gan 97bb85d505 [Model] GraphSAGE with control variate sampling on new sampler (#1355) 
* control variate first commit

* bug fixes

* split to single and multi GPU

* update readme

* bugfix

* bugfix

* remove push

* bugfix on multi gpu

* update README

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2020-04-26 22:06:21 +08:00
Minjie Wang d876680a36 [Model][Perf] Improve sage sampling performance (#1364) 
* improve speed

* fix bugs

* upd reg test
 2020-03-15 23:20:30 +08:00
Zihao Ye 650f6ee1e0 [NN] Add commonly used GNN models from examples to dgl.nn modules. (#748) 
* gat

* upd

* upd sage

* upd

* upd

* upd

* upd

* upd

* add gmmconv

* upd ggnn

* upd

* upd

* upd

* upd

* add citation examples

* add README

* fix cheb

* improve doc

* formula

* upd

* trigger

* lint

* lint

* upd

* add test for transform

* add test

* check

* upd

* improve doc

* shape check

* upd

* densechebconv, currently not correct (?)

* fix cheb

* fix

* upd

* upd sgc-reddit

* upd

* trigger
 2019-08-27 18:21:19 +08:00
Chao Ma f99725adbc all demo use python-3 (#555) 2019-05-23 10:38:05 +08:00
hbsun2113 8c750170e8 [Model][Pytorch] GraphSAGE (#403) 
* simple implemention of GraphSAGE

* update the abstract method and `torch.nn` modules are more utilized
 2019-02-22 15:48:53 -05:00