- 16 Jun, 2021 3 commits
-
-
Hengrui Zhang authored
Co-authored-by:zhjwy9343 <6593865@qq.com>
-
Hengrui Zhang authored
Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
espylapiza authored
* [Example] GCN on ogbn-arxiv dataset * Add README.md * Update GCN implementation on ogbn-arxiv * Update GCN on ogbn-arxiv * Fix typo * Use evaluator to get results * Fix duplicated * Fix duplicated * Update GCN on ogbn-arxiv * Add GAT for ogbn-arxiv * Update README.md * Update GAT on ogbn-arxiv * Update README.md * Update README * Update README.md * Update README.md * Add GAT implementation for ogbn-proteins * Add GAT implementation for ogbn-proteins * Add GAT implementation for ogbn-proteins * Add GAT implementation for ogbn-proteins * Add GAT implementation for ogbn-proteins * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Update examples for ogbn-arxiv * Add examples for ogbn-products. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-proteins. * Update examples for ogbn-products. * Update examples for ogbn-products. * Update examples for ogbn-arxiv. * Update examples for ogbn-arxiv. * Update examples for ogbn-arxiv. * Update examples for ogbn-proteins. * Update examples for ogbn-products. * Update examples for ogbn-proteins. * Update examples for ogbn-arxiv. * Update examples for ogbn-proteins. * Update README.md * [Example] ogbn-arxiv & ogbn-proteins * [Example] ogbn-arxiv & ogbn-proteins * [Example] ogbn-arxiv & ogbn-proteins * [Example] ogbn-arxiv & ogbn-proteins Co-authored-by:
Mufei Li <mufeili1996@gmail.com> Co-authored-by:
Zihao Ye <expye@outlook.com>
-
- 15 Jun, 2021 1 commit
-
-
Hengrui Zhang authored
* reformat and update performance * Update README.md * Fix typos Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
- 13 Jun, 2021 2 commits
-
-
nv-dlasalle authored
[Performance] Perform to_block on the GPU when the dataloader is created with a GPU `device`. (#3016) * add output device for dataloading * Update dataloader * Get sampler device from dataloader * Fix line length * Update examples * Fix to_block GPU for empty relation types * Handle the case where the DistGraph has None for the underlying graph Co-authored-by:Da Zheng <zhengda1936@gmail.com>
-
nv-dlasalle authored
-
- 11 Jun, 2021 1 commit
-
-
nv-dlasalle authored
* Split from NCCL PR * Fix type in comment * Expand documentation for sparse_all_to_all_push * Restore previous behavior in example * Re-work optimizer to use NCCL based on gradient location * Allow for running with embedding on CPU but using NCCL for gradient exchange * Optimize single partition case * Fix pylint errors * Add missing include * fix gradient indexing * Fix line continuation * Migrate 'first_step' * Skip tests without enough GPUs to run NCCL * Improve empty tensor handling for pytorch 1.5 * Fix indentation * Allow multiple NCCL communicator to coexist * Improve handling of empty message * Update python/dgl/nn/pytorch/sparse_emb.py Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com> * Update python/dgl/nn/pytorch/sparse_emb.py Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com> * Keepy empty tensor dimensionaless * th.empty -> th.tensor * Preserve shape for empty non-zero dimension tensors * Use shared state, when embedding is shared * Add support for gathering an embedding * Fix typo * Fix more typos * Fix backend call * Use NodeDataLoader to take advantage of ddp * Update training script to share memory * Only squeeze last dimension * Better handle empty message * Keep embedding on the target device GPU if dgl_sparse if false in RGCN example * Fix typo in comment * Add asserts * Improve documentation in example Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com>
-
- 10 Jun, 2021 1 commit
-
-
nv-dlasalle authored
* Make multigpu graphsage use whole datset * Specify queeze dimension * Remove squeeze dimension Co-authored-by:Quan (Andy) Gan <coin2028@hotmail.com>
-
- 07 Jun, 2021 1 commit
-
-
Tong He authored
* reorg and expose dgl.geometry * fix lint * fix test * fix
-
- 04 Jun, 2021 1 commit
-
-
Alex P authored
-
- 03 Jun, 2021 1 commit
-
-
KounianhuaDu authored
* tahin * readme * readme * readme * readme * readme * readme * main * main * new_line * update Co-authored-by:zhjwy9343 <6593865@qq.com>
-
- 02 Jun, 2021 1 commit
-
-
Moritz Blum authored
corrected docstring, the model uses two R-GCN layers Co-authored-by:Minjie Wang <wmjlyjemaine@gmail.com>
-
- 26 May, 2021 1 commit
-
-
Da Zheng authored
* explicitly set the graph format. * fix. * fix. * fix launch script. * fix readme. Co-authored-by:
Zheng <dzzhen@3c22fba32af5.ant.amazon.com> Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com> Co-authored-by:
Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
-
- 17 May, 2021 2 commits
-
-
yifeim authored
* add lda model * tweak latent dirichlet allocation * Update README.md * Update README.md * update example index * update header * minor tweak * add example test * update doc * Update README.md * Update README.md * add partial_fit for free * Update examples/pytorch/lda/lda_model.py Co-authored-by:
Quan (Andy) Gan <coin2028@hotmail.com> * Update examples/pytorch/lda/example_20newsgroups.py Co-authored-by:
Quan (Andy) Gan <coin2028@hotmail.com> * Update lda_model.py * bugfix torch Gamma uses rate parameter Co-authored-by:
Yifei Ma <yifeim@amazon.com> Co-authored-by:
Quan (Andy) Gan <coin2028@hotmail.com>
-
Quan (Andy) Gan authored
-
- 14 May, 2021 1 commit
-
-
Quan (Andy) Gan authored
* standardizing thread_wrapped_func * lints * Update __init__.py
-
- 12 May, 2021 1 commit
-
-
xnouhz authored
* [example] label propagation and correct&smooth * update * update * update * update * [docs] add speed for c&s * update * [fix] remove gat&consistent with the author's code * [feat] multi-type adj norm supported * update Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
- 11 May, 2021 2 commits
-
-
lt610 authored
* graphsaint * graphsaint * graphsaint * graphsaint * fixed the model * fixed some bugs * fixed the computing of normalization and updated the results * fixed some bugs and updated the results * Update utils.py * Update train_sampling.py * Update train_sampling.py Co-authored-by:
Mufei Li <mufeili1996@gmail.com> Co-authored-by:
Quan (Andy) Gan <coin2028@hotmail.com>
-
Quan (Andy) Gan authored
* pytorch lightning initial examples * revert most changes in dataloader to favor #2886. * address comments
-
- 03 May, 2021 1 commit
-
-
xiang song(charlie.song) authored
* Draft for sparse emb * add some notes * Fix * Add sparse optim for dist pytorch * Update test * Fix * upd * upd * Fix * Fix * Fix bug * add transductive exmpale * Fix example * Some fix * Upd * Fix lint * lint * lint * lint * upd * Fix lint * lint * upd * remove dead import * update * lint * update unitest * update example * Add adam optimizer * Add unitest and update data * upd * upd * upd * Fix docstring and fix some bug in example code * Update rgcn readme Co-authored-by:
Ubuntu <ubuntu@ip-172-31-57-25.ec2.internal> Co-authored-by:
Ubuntu <ubuntu@ip-172-31-24-210.ec2.internal> Co-authored-by:
Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
-
- 27 Apr, 2021 1 commit
-
-
Hengrui Zhang authored
* [Example] add implementation of grace * [Doc] add grace in the index file * fix * fix typos Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
- 26 Apr, 2021 1 commit
-
-
Mufei Li authored
-
- 25 Apr, 2021 1 commit
-
-
xnouhz authored
* [example] deepergcn * update * update * update * update * update Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
- 23 Apr, 2021 2 commits
-
-
Chen Sirui authored
* Ready for PR * refractor code Co-authored-by:
Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal> Co-authored-by:
Tianjun Xiao <xiaotj1990327@gmail.com>
-
Chen Sirui authored
* Add DCRNN Example * Bugfix DCRNN * Bugfix DCRNN * Bugfix Train/eval have different flow * Performance Matched * modified tgn for research * kind of fixed 2hop issue * remove data * bugfix and improve the performance * Refractor Code Co-authored-by:
Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal> Co-authored-by:
Tianjun Xiao <xiaotj1990327@gmail.com>
-
- 21 Apr, 2021 1 commit
-
-
Tingzhang Zhao authored
Change the format of README.md
-
- 20 Apr, 2021 1 commit
-
-
Quan (Andy) Gan authored
Co-authored-by:
Minjie Wang <wmjlyjemaine@gmail.com> Co-authored-by:
Jinjing Zhou <VoVAllen@users.noreply.github.com>
-
- 19 Apr, 2021 1 commit
-
-
Tingzhang Zhao authored
* Update README.md Add description of RECT * [Example] Add implementation of RECT [Example] Add implementation of RECT * Update classify.py Modify the class names and the function names mentioned above * Update main.py Modify the function names mentioned above * Update label_utils.py Adjust the comments * Update README.md Add the github information Co-authored-by:Tianjun Xiao <xiaotj1990327@gmail.com>
-
- 15 Apr, 2021 1 commit
-
-
nv-dlasalle authored
* Start on uniform GPU sampling * Save more work * Get cu file compiling * Update sampling * More changes * Get GPU sampling for uniform probabilities solved * Fix batch tensor migration * Fix * update kernels * expand blocking * Undo testing change * Cut down on sampling overhead * Fix replacement * Update unit tests * Add option to gpu sample in graphsage * Copy only csc to gpu * Add ogbn support * Fix linting * Remove nvtx from sample * Improve documentation and error checking * Expand documentation * Update assert checking * delete extra space * Use standard dataloader when dataset is a dictionary * ogb -> ogbn * Fix edge selection determinism * Fix typos * Remove nvtx * Add comment for self.fanout_arrays and assert * Fix linting * Migrate to scalarbatcher * Fix indentation * Fix batcher * Fix indexing * Only use databatcher for GPU * Convert to DGL NDArray to PyTorch Tensor * Add optimization for PyTorch's F.tensor() for list of GPU tensors Co-authored-by:Da Zheng <zhengda1936@gmail.com>
-
- 13 Apr, 2021 1 commit
-
-
maqy authored
* multi gpus training for mag240m * remove unused function * small changes * more fix * use multi gpus to eval and more fix Co-authored-by:Quan Gan <coin2028@hotmail.com>
-
- 12 Apr, 2021 1 commit
-
-
Chen Sirui authored
* Add hgat example * Add experiment * Clean code * clear the code * Add index in README * Add index in README * Add index in README * Add index in README * Add index in README * Add index in README * Change the code title and folder name * Ready to merge * Prepare for rebase and change message passing function * use git ignore to handle empty file * change file permission to resolve empty file * Change permission * change file mode * Finish Coding * working code cpu * pyg compare * Accelerate with batching * FastMode Enabled * update readme * Update README.md * refractor code * add graphsim code * modified code * few fix * Modified graphsim * Simple Model Added * Clean up code * Refractor the code for Merge * Bugfix enable gradient when train * update readme and format Co-authored-by:
Chen <chesirui@3c22fbe5458c.ant.amazon.com> Co-authored-by:
Tianjun Xiao <xiaotj1990327@gmail.com> Co-authored-by:
Ubuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal> Co-authored-by:
Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal>
-
- 11 Apr, 2021 1 commit
-
-
TianleZhang authored
int has no attribute of item Co-authored-by:Quan (Andy) Gan <coin2028@hotmail.com>
-
- 08 Apr, 2021 1 commit
-
-
Da Zheng authored
Co-authored-by:
Ubuntu <ubuntu@ip-172-31-73-81.ec2.internal> Co-authored-by:
Jinjing Zhou <VoVAllen@users.noreply.github.com>
-
- 07 Apr, 2021 2 commits
-
-
xnouhz authored
* [example] jknet * update * update * update Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
Ruoyun Huang authored
* Typo in command example. process_nowplaying_rs.py is the right name under this folder. * Update README.md * Update README.md Co-authored-by:Quan (Andy) Gan <coin2028@hotmail.com>
-
- 06 Apr, 2021 1 commit
-
-
Sergey Ivanov authored
* Add BGNN example * Update bgnn example * Remove uncessary files * update model index * style fix * Add Readme for BGNN example * Update documentation; remove preprocessing Co-authored-by:
s.ivanov <s.ivanov@mesos-slave080-am6.central.criteo.prod> Co-authored-by:
s.ivanov <s.ivanov@mesos-slave150-am6.central.criteo.prod> Co-authored-by:
Quan Gan <coin2028@hotmail.com> Co-authored-by:
s.ivanov <s.ivanov@mesos-slave148-am6.central.criteo.prod> Co-authored-by:
s.ivanov <s.ivanov@mesos-slave078-am6.central.criteo.prod> Co-authored-by:
Minjie Wang <wmjlyjemaine@gmail.com>
-
- 05 Apr, 2021 2 commits
-
-
KounianhuaDu authored
* compgcn * readme * readme * update * readme Co-authored-by:zhjwy9343 <6593865@qq.com>
-
Quan (Andy) Gan authored
* RGAT baseline * update links (data.dgl.ai hasn't refreshed yet) * nvm still use dgl.ai domain * descriptions * actually it took less * address comments * stop worrying about cache * update links * oops
-
- 03 Apr, 2021 1 commit
-
-
Yu Sun authored
* [Example]Variational Graph Auto-Encoders * change dgl dataset to single directional graph * clean code * refresh * fix bug * fix bug * fix bug * add gpu Co-authored-by:
Tianjun Xiao <xiaotj1990327@gmail.com> Co-authored-by:
Minjie Wang <wmjlyjemaine@gmail.com>
-
- 30 Mar, 2021 1 commit
-
-
Da Zheng authored
* remove num_workers. * remove num_workers. * remove num_workers. * remove num-servers. * update error message. * update docstring. * fix docs. * fix tests. * fix test. * fix. * print messages in test. * fix. * fix test. * fix. Co-authored-by:Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
-