文件历史

641 次代码提交

作者 SHA1 备注 提交日期
Rhett Ying 0bd6bc80cd [examples]educe memory consumption (#4558) (#4596)
* [examples]educe memory consumption

* reffine help message

* refine
2022-09-20 15:59:41 +08:00
Chang Liu fce9614089 [Example][Refactor] Refactor graphsage multigpu and full-graph example (#4430)
* Add refactors for multi-gpu and full-graph example

* Fix format

* Update

* Update

* Update
2022-09-05 09:05:56 +08:00
rudongyu b9290e8b2d [Example] SEAL for OGBL (#4291)
* [Example] SEAL for OGBL

* update index

* update

* fix readme typo

* add seal sampler

* modify set ops

* prefetch

* efficiency test

* update

* optimize

* fix ScatterAdd dtype issue

* update sampler style

* update

Co-authored-by: Quan Gan <coin2028@hotmail.com>
2022-08-31 11:48:07 +08:00
Zhiteng Li 97b2ab53e2 [Example] directional_GSN for ogbg-molpcba (#4405)
* version-1

* version-2

* version-3

* update examples/README

* Update .gitignore

* update performance in README, delete scripts

* 1st approving review

* 2nd approving review

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-08-26 15:37:54 +08:00
Chang Liu 40a2f3c760 [Example][Refactor] Refactor RGCN example (#4327)
* Refactor full graph entity classification

* Refactor rgcn with sampling

* README update

* Update

* Results update

* Respect default setting of self_loop=false in entity.py

* Update

* Update README

* Update for multi-gpu

* Update
2022-08-25 13:19: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
Rhett Ying e14860d9cc [Example][Bug] fix device index for dist train on GPUs (#4403) 2022-08-18 13:00:21 +08:00
Ereboas 49c8179548 [Example] NGNN for ogbl (#4328)
* NGNN for ogbl

* modify doc organization.

* merge similar parts

* 1st approving review.

* minor changes

* Remove the "Usage" section.

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-08-13 19:11:45 +08:00
YJ-Zhao a88e7f7e28 [Example]rgcn-ogbn-mag (#4331)
* rgcn-ogbn-mag

* Add link in README.md

* correct code-format,add the reset_parameters function to the HeteroEmbedding module

* add the annotation in hetero.py

* add a unit test

* modify format

* Update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-50-143.us-west-2.compute.internal>
2022-08-10 20:12:31 +08:00
Chang Liu 9a16a5e03d [Example][Refactor] Refactor GIN example (#4280)
* Refactor GIN example

* Update

* Update README

* Minor update

* README update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-08-01 13:40:28 +08:00
Wey Gu ee91863eb0 [Example] Fixed device type in GraphSAGE inference (#4306)
* Fixed device type in inference

* change buffer_device instead
2022-07-28 13:25:09 +08:00
Chang Liu 4f7972957b [Bugfix] Fix dataloader pytorch cuda indexing (#4297)
* Modify to repro crash

* Revert to orig. scenario and add fix

* Update

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-26 16:59:58 +08:00
Chang Liu 6e1be69a84 [Example][Refactor] Refactor GAT example (#4240)
* Refactor gat example

* Add ppi support

* Minor update

* Update

* Update

* Change valid_xxx to val_xxx

* Readme Update

* Update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-07-22 16:48:46 +08:00
Tianyue Cao 05aca98d6a [Example] Progressive Sample Selection (PSS). (#4263)
* upload PSS

* upload PSS

* upload PSS

* pss code reformat

* fix bug

* update README

* update train bash

* remove vit

* update README

* delete InfoPlotter

* delete Smooth_AP_loss.py

* update README

* update README

Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
2022-07-21 08:08:33 +08:00
Chang Liu 701b746b82 [Example][Bugfix] graphsage node classification example (#4260)
* Fix node_classification.py

* Minor update

Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-07-19 11:28:35 -07:00
Chang Liu e26d206437 [Example][Bugfix] Fix link pred example in graphsage (#4255) 2022-07-14 08:30:51 +08:00
Chang Liu c56e27a807 [Bugfix] Accessing data from the indexes stored in same device (#4242)
* First update to fix two examples

* Update to fix RGCN/graphsage example and dataloader

* Update
2022-07-13 09:02:07 +08:00
Xin Yao 3e26c3d198 fix caregnn (#4211)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-07-08 16:16:30 +08:00
Chang Liu 52d43127f7 [Example][Bugfix] Fix arma example (#4218)
* Fix arma example

* Update

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-08 12:25:21 +08:00
Chang Liu 28b0904779 [Example][Bugfix] Fix dimenet example (#4219)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-07 14:52:10 +08:00
Chang Liu 0d1b2c5a2c [Example] Update README file of eges example (#4220) 2022-07-07 09:14:40 +08:00
Chang Liu 885be1784b [Example][Refactor] Refactor GCN example (#4160)
* Refactor GCN example

* Refactor GCN based on graphsage

* Readme update

* Minor update

* update

* Remove user-defined GCN implementation

* README update

* Update

* Update CONTRIBUTORS.md

* update task_example_test

Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-07-05 15:58:31 +08:00
Chang Liu 0f0e7c7fad [Example][Bugfix] Bugfix for dgi example (#4201) 2022-07-02 12:48:17 +08:00
Chang Liu a9768cb38f [Example][Refactor] Minor update on the golden example (#4197)
* minor update on golden example

* update

* update

* Update README

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-07-01 14:34:45 +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
Chang Liu a6bd96aa96 Fix example crashes due to DGL API update (#4194)
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-30 21:10:44 +08:00
Lucas Prieto 548c85fff6 [BugFix] Fix Correct&Smooth (#4102) (#4158)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Xin Yao <xiny@nvidia.com>
2022-06-23 18:11:15 +08:00
Chang Liu 598d746e59 [Example][Bugfix] Remove all torchtext legacy-related APIs for pytorch/pinsage example (#4130)
* Remove all torchtext legacy-related APIs

* Remove unused BagOfWordsPretrained class, and fix some typos

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-23 16:06:18 +08:00
Xin Yao 077e002fe5 [Bugfix][Rework] Automatically unpin tensors pinned by DGL (rework #3997) (#4135)
* Explicitly unpin tensoradapter allocated arrays

* Undo unrelated change

* Add unit test

* update unit test

* add pinned_by_dgl flag to NDArray::Container

* use dgl.ndarray for holding the pinning status

* update multi-gpu uva inference

* reinterpret cast NDArray::Container* to DLTensor* in MoveAsDLTensor

* update unpin column and examples

* add unit test for unpin column

Co-authored-by: Dominique LaSalle <dlasalle@nvidia.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-06-23 13:56:54 +08:00
Chang Liu 549df65a77 [Bugfix] Fix example case: examples/pytorch/ogb/ogbn-proteins and examples/pytorch/ogb/ogbn-products (#4080)
* [Bugfix] Fix ogbn-gat-proteins/products examples

* Remove unused BatchSampler definition

* Remove comments to ease reading/reviewing

* Remove dataloader wrapper
2022-06-09 15:45:28 +08:00
Sai Kandregula a3ea4873d3 [Example] Remove Bias in Linear layer because of BatchNorm1d usage (#4071)
* typo fix in TUDataset docs

* remove bias in Linear layer because of BatchNorm1d

Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-04 17:16:20 +08:00
Chang Liu d31448dd24 [Bugfix] Fix GCMC examples (#4082)
* [Example][Bug] Fix GCMC examples

* Revert the change of model.py from #74f01405
2022-06-04 11:19:33 +08:00
RecLusIve-F 0f2ff47de6 [Example] Add BGRL example (#4077)
* Add BGRL example

* Update README.md

* Update utils.py

* Update

* Update utils.py
2022-06-03 23:13:30 +08:00
Chang Liu 9e46423e4f [Bugfix] Fix cluster-gat examples (#4068)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-06-02 21:39:59 +08:00
Mufei Li bef993076e [Bugfix] Fix PinSAGE Benchmark (#4058)
* Update

* Update

* Update dgl.data.rst

* CI
2022-05-28 00:31:57 +08:00
彭齐荣 d1124b7bef [Example][BugFix] Fix Lightning GraphSAGE examples (dmlc#4046) (#4053) 2022-05-26 12:40:40 +08:00
Quan (Andy) Gan 744896e2d8 fix examples (#4016)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-05-23 23:41:12 +08:00
Mufei Li bdaccc8270 [Bug Fix] Fix KeyError for rgcn-hetero Entity Classification w/ minibatch training (#4022)
* Update model.py

* Update entity_classify_mb.py
2022-05-23 20:25:01 +08:00
rudongyu 3fe5eea791 [NN] Label Propagation & Directional Graph Networks (#4017)
* add label propagation module

* fix prev bug in example

* add dgn

* fix linting and doc issues

* update label propagation & dgn

* update label propagation & dgn

* update example

* fix unit test

* fix agg heritage issue

* fix agg issue

* fix lint

* fix idx

* fix lp gpu issue

* Update

* Update

Co-authored-by: mufeili <mufeili1996@gmail.com>
2022-05-20 15:23:19 +08:00
Hao Yuan 6de7d5faa7 [Bugfix] Fix conditional judgment of cuda (#4026) (#4027) 2022-05-20 11:01:03 +08:00
Peiqi Yin cba465f2c8 [BugFix] hetero ogbn-mag full neighbor inference. (#4010)
* fix hetero ogbn-mag full neighbor inference.

* fix hetero

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-05-18 16:46:55 +08:00
Quan (Andy) Gan 74f0140533 [Example] Rename NodeDataLoader to DataLoader in GraphSAGE example (#3972)
* rename

* Update node_classification.py

* more fixes...

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-05-11 09:10:54 +08:00
Xinger d4d4fa1eaf Update pinsage readme: torchtext>=0.9.0 (#3974)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-05-10 12:41:01 +08:00
Hengrui Zhang ae7e3db631 fix bugs in gat training script (#3505)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2022-04-29 17:53:00 +08:00
Rhett Ying 37be02a486 [Feature] enable socket net_type for rpc (#3951)
* [Feature] enable socket net_type for rpc

* fix lint

* fix lint

* fix build issue on windows

* fix test failure on windows

* fix test failure

* fix cpp unit test failure

* net_type blocking max_try_times

* fix other comments

* fix lint

* fix comment

* fix lint

* fix cpp
2022-04-28 00:10:06 +08:00
Sharique Shamim 306e0a46e1 [Bug] Fixed optimizer definition in gnn_explainer (#3944) (#3948)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-04-25 16:39:44 +08:00
Quan (Andy) Gan dc1c0ac28a [Example][Bug] Fix GAT example (#3947)
* fix

* oops

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-04-22 23:44:45 +08:00
WangYQ f5bba284f8 [Example] Add EGES example (#3756)
* add eges example

* remove csv files and add data link

* Update README.md

* Update main.py

* Update model.py

* Update sampler.py

* Update utils.py

* Update model.py

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-14 16:23:08 +08:00
Tong He a0bf5daa65 fix (#3903)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-04-13 12:28:19 +08:00
Mufei Li c380f0b5fa Update gatv2.py (#3921)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-13 09:47:52 +08:00