1. 08 May, 2020 1 commit
  2. 06 May, 2020 1 commit
    • Linfang He's avatar
      [Model] GATNE-T (#1470) · 6ae440db
      Linfang He authored
      
      
      * Add an example for GATNE-T
      
      * Update README.md
      
      * Add links for datasets
      
      * Update README.md with running results
      
      * Update README.md
      
      * Add comments
      
      * Update main.py
      
      * Remove node type name `user`
      Co-authored-by: default avatarDa Zheng <zhengda1936@gmail.com>
      6ae440db
  3. 03 May, 2020 1 commit
    • Da Zheng's avatar
      [Feature] Distributed graph store (#1383) · 2190c39d
      Da Zheng authored
      * initial version from distributed training.
      
      This is copied from multiprocessing training.
      
      * modify for distributed training.
      
      * it's runnable now.
      
      * measure time in neighbor sampling.
      
      * simplify neighbor sampling.
      
      * fix a bug in distributed neighbor sampling.
      
      * allow single-machine training.
      
      * fix a bug.
      
      * fix a bug.
      
      * fix openmp.
      
      * make some improvement.
      
      * fix.
      
      * add prepare in the sampler.
      
      * prepare nodeflow async.
      
      * fix a bug.
      
      * get id.
      
      * simplify the code.
      
      * improve.
      
      * fix partition.py
      
      * fix the example.
      
      * add more features.
      
      * fix the example.
      
      * allow one partition
      
      * use distributed kvstore.
      
      * do g2l map manually.
      
      * fix commandline.
      
      * a temp script to save reddit.
      
      * fix pull_handler.
      
      * add pytorch version.
      
      * estimate the time for copying data.
      
      * delete unused code.
      
      * fix a bug.
      
      * print id.
      
      * fix a bug
      
      * fix a bug
      
      * fix a bug.
      
      * remove redundent code.
      
      * revert modify in sampler.
      
      * fix temp script.
      
      * remove pytorch version.
      
      * fix.
      
      * distributed training with pytorch.
      
      * add distributed graph store.
      
      * fix.
      
      * add metis_partition_assignment.
      
      * fix a few bugs in distributed graph store.
      
      * fix test.
      
      * fix bugs in distributed graph store.
      
      * fix tests.
      
      * remove code of defining DistGraphStore.
      
      * fix partition.
      
      * fix example.
      
      * update run.sh.
      
      * only read necessary node data.
      
      * batching data fetch of multiple NodeFlows.
      
      * simplify gcn.
      
      * remove unnecessary code.
      
      * use the new copy_from_kvstore.
      
      * update training script.
      
      * print time in graphsage.
      
      * make distributed training runnable.
      
      * use val_nid.
      
      * fix train_sampling.
      
      * add distributed training.
      
      * add run.sh
      
      * add more timing.
      
      * fix a bug.
      
      * save graph metadata when partition.
      
      * create ndata and edata in distributed graph store.
      
      * add timing in minibatch training of GraphSage.
      
      * use pytorch distributed.
      
      * add checks.
      
      * fix a bug in global vs. local ids.
      
      * remove fast pull
      
      * fix a compile error.
      
      * update and add new APIs.
      
      * implement more methods in DistGraphStore.
      
      * update more APIs.
      
      * rename it to DistGraph.
      
      * rename to DistTensor
      
      * remove some unnecessary API.
      
      * remove unnecessary files.
      
      * revert changes in sampler.
      
      * Revert "simplify gcn."
      
      This reverts commit 0ed3a34ca714203a5b45240af71555d4227ce452.
      
      * Revert "simplify neighbor sampling."
      
      This reverts commit 551c72d20f05a029360ba97f312c7a7a578aacec.
      
      * Revert "measure time in neighbor sampling."
      
      This reverts commit 63ae80c7b402bb626e24acbbc8fdfe9fffd0bc64.
      
      * Revert "add timing in minibatch training of GraphSage."
      
      This reverts commit e59dc8957a414c7df5c316f51d78bce822bdef5e.
      
      * Revert "fix train_sampling."
      
      This reverts commit ea6aea9a4aabb8ba0ff63070aa51e7ca81536ad9.
      
      * fix lint.
      
      * add comments and small update.
      
      * add more comments.
      
      * add more unit tests and fix bugs.
      
      * check the existence of shared-mem graph index.
      
      * use new partitioned graph storage.
      
      * fix bugs.
      
      * print error in fast pull.
      
      * fix lint
      
      * fix a compile error.
      
      * save absolute path after partitioning.
      
      * small fixes in the example
      
      * Revert "[kvstore] support any data type for init_data() (#1465)"
      
      This reverts commit 87b6997b
      
      .
      
      * fix a bug.
      
      * disable evaluation.
      
      * Revert "Revert "[kvstore] support any data type for init_data() (#1465)""
      
      This reverts commit f5b8039c6326eb73bad8287db3d30d93175e5bee.
      
      * support set and init data.
      
      * support set and init data.
      
      * Revert "Revert "[kvstore] support any data type for init_data() (#1465)""
      
      This reverts commit f5b8039c6326eb73bad8287db3d30d93175e5bee.
      
      * fix bugs.
      
      * fix unit test.
      
      * move to dgl.distributed.
      
      * fix lint.
      
      * fix lint.
      
      * remove local_nids.
      
      * fix lint.
      
      * fix test.
      
      * remove train_dist.
      
      * revert train_sampling.
      
      * rename funcs.
      
      * address comments.
      
      * address comments.
      
      Use NodeDataView/EdgeDataView to keep track of data.
      
      * address comments.
      
      * address comments.
      
      * revert.
      
      * save data with DGL serializer.
      
      * use the right way of getting shape.
      
      * fix lint.
      
      * address comments.
      
      * address comments.
      
      * fix an error in mxnet.
      
      * address comments.
      
      * add edge_map.
      
      * add more test and fix bugs.
      Co-authored-by: default avatarZheng <dzzhen@186590dc80ff.ant.amazon.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-6-131.us-east-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-26-167.us-east-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-16-150.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-16-250.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-30-135.us-west-2.compute.internal>
      2190c39d
  4. 28 Apr, 2020 2 commits
  5. 26 Apr, 2020 1 commit
  6. 24 Apr, 2020 1 commit
  7. 13 Apr, 2020 1 commit
  8. 09 Apr, 2020 1 commit
  9. 05 Apr, 2020 1 commit
  10. 28 Mar, 2020 1 commit
  11. 27 Mar, 2020 1 commit
    • Minjie Wang's avatar
      [NN] Add HeteroGraphConv module for cleaner module definition (#1385) · 3efb5d8e
      Minjie Wang authored
      * Add HeteroGraphConv
      
      * add custom aggregator; some docstring
      
      * debugging
      
      * rm print
      
      * fix some acc bugs
      
      * fix initialization problem in weight basis
      
      * passed tests
      
      * lint
      
      * fix graphconv flag; add error message
      
      * add mxnet heteroconv
      
      * more fix for mx
      
      * lint
      
      * fix torch cuda test
      
      * fix mx test_nn
      
      * add exhaust test for graphconv
      
      * add tf heteroconv
      
      * fix comment
      3efb5d8e
  12. 20 Mar, 2020 1 commit
  13. 18 Mar, 2020 1 commit
  14. 16 Mar, 2020 1 commit
    • Minjie Wang's avatar
      [Model] RGCN mini-batch training; Better bipartite graph support (#1337) · 6111ea46
      Minjie Wang authored
      * change the model to use node embedding only
      
      * minibatch training
      
      * add readme
      
      * small fix
      
      * backward-compatible
      
      * backward compatible
      
      * modify to_block; rgcn changed
      
      * fix
      
      * fix transform
      
      * fix bug in unittest script
      
      * docstring
      
      * fix lint
      
      * add tests
      
      * address comments; fix offline eval
      
      * gitignore
      6111ea46
  15. 15 Mar, 2020 1 commit
  16. 13 Mar, 2020 1 commit
  17. 10 Mar, 2020 1 commit
  18. 07 Mar, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Model][Sampler] GraphSAGE model, bipartite graph conversion & remove edges API (#1297) · a9520f71
      Quan (Andy) Gan authored
      * remove edge and to bipartite and graphsage with sampling
      
      * fixes
      
      * fixes
      
      * fixes
      
      * reenable multigpu training
      
      * fixes
      
      * compatibility in DGLGraph
      
      * rename to compact_as_bipartite
      
      * bugfix
      
      * lint
      
      * add offline inference
      
      * skip GPU tests
      
      * fix
      
      * addresses comments
      
      * fix
      
      * fix
      
      * fix
      
      * more tests
      
      * more docs and unit tests
      
      * workaround for empty slice on empty data
      a9520f71
  19. 05 Mar, 2020 3 commits
    • Mufei Li's avatar
      Update (#1319) · 4ec8f204
      Mufei Li authored
      4ec8f204
    • Zihao Ye's avatar
      [Refactor] Unify DGLGraph, BatchedDGLGraph and DGLSubGraph (#1216) · 93ac29ce
      Zihao Ye authored
      
      
      * upd
      
      * upd
      
      * upd
      
      * lint
      
      * fix
      
      * fix test
      
      * fix
      
      * fix
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd tutorial
      
      * upd
      
      * upd
      
      * fix kg
      
      * upd doc organization
      
      * refresh test
      
      * upd
      
      * refactor doc
      
      * fix lint
      Co-authored-by: default avatarMinjie Wang <minjie.wang@nyu.edu>
      93ac29ce
    • Tong He's avatar
      [Model] Scene Graph Extraction Model with GluonCV (#1260) · cbee4278
      Tong He authored
      
      
      * add working scripts
      
      * add frcnn training script
      
      * remove redundent files
      
      * refactor validation computation, will optimize sgdet and training
      
      * validation finally finished
      
      * f-rcnn training
      
      * test reldn
      
      * rm file
      
      * update reldn training
      
      * data preprocess to h5
      
      * temp
      
      * use coco json
      
      * fix conflict
      
      * new obj dataset for detection
      
      * update training
      
      * before cleanup
      
      * remove abundant files
      
      * add arg parse to train
      
      * cleanup code file
      
      * update
      
      * fix
      
      * add readme
      
      * add ipynb as demo
      
      * add demo pic
      
      * update readme
      
      * add demo script
      
      * improve paths
      
      * improve readme
      
      * add docstrings
      
      * fix args description
      
      * update readme
      
      * add models from s3
      
      * update README
      Co-authored-by: default avatarMinjie Wang <minjie.wang@nyu.edu>
      cbee4278
  20. 04 Mar, 2020 1 commit
  21. 24 Feb, 2020 1 commit
  22. 23 Feb, 2020 1 commit
    • Chao Ma's avatar
      [KVStore] Small fix on kvstore (#1284) · ddf8c858
      Chao Ma authored
      * update
      
      * str to int
      
      * update
      
      * update
      
      * update
      
      * test remove sync
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * clear
      ddf8c858
  23. 17 Feb, 2020 1 commit
    • Chao Ma's avatar
      [KVStore] New kvstore used by DGL-KE (#1263) · b133abb8
      Chao Ma authored
      * new kvstore
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * test warning
      
      * update
      
      * update
      
      * udpate
      
      * update
      
      * update
      
      * update
      
      * update
      
      * small fix
      
      * small fix
      
      * get group count
      
      * update
      
      * update
      
      * make file
      
      * update
      
      * use addr
      
      * get id
      
      * partition book
      
      * update
      
      * partition
      
      * barrier
      
      * update
      
      * loop count
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * add mxnet demo
      
      * update ip
      
      * update
      
      * update
      
      * update
      
      * random
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * fix lint
      
      * fix lint
      b133abb8
  24. 13 Feb, 2020 1 commit
  25. 08 Feb, 2020 1 commit
  26. 07 Feb, 2020 1 commit
  27. 03 Feb, 2020 1 commit
    • Mufei Li's avatar
      [DGL-LifeSci] Migration and Refactor (#1226) · 828a5e5b
      Mufei Li authored
      * First commit
      
      * Update
      
      * Update splitters
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Migrate ACNN
      
      * Fix
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Finish classification
      
      * Update
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * Fix
      
      * Fix
      
      * Fix
      
      * Update
      
      * Update
      
      * Update
      
      * trigger CI
      
      * Fix CI
      
      * Update
      
      * Update
      
      * Update
      
      * Add default values
      
      * Rename
      
      * Update deprecation message
      828a5e5b
  28. 31 Jan, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] Random Walk for 0.5 (#1209) · a9dabcc7
      Quan (Andy) Gan authored
      * trying to refactor IndexSelect
      
      * partial implementation
      
      * add index select and assign for floats as well
      
      * move to random choice source
      
      * more updates
      
      * fixes
      
      * fixes
      
      * more fixes
      
      * adding python impl
      
      * fixes
      
      * unit test
      
      * lint
      
      * lint x2
      
      * lint x3
      
      * update metapath2vec
      
      * debugging performance
      
      * still debugging for performance
      
      * tuning
      
      * switching to succvec
      
      * redo
      
      * revert non-uniform sampler to use vector
      
      * still not fast
      
      * why does this crash with OpenMP???
      
      * because there was a data race!!!
      
      * add documentations and remove assign op
      
      * lint
      
      * lint x2
      
      * lol what have i done
      
      * lint x3
      
      * fix and disable gpu testing
      
      * bugfix
      
      * generic random walk
      
      * reorg the random walk source code
      
      * Update randomwalks.h
      
      * Update randomwalks_cpu.cc
      
      * rename file
      
      * move internal function to anonymous ns
      
      * reorg & docstrings
      
      * constant restart probability
      
      * docstring fix
      
      * more commit
      
      * random walk with restart, tested
      
      * some fixes
      
      * switch to using NDArray for choice
      
      * massive fix & docstring
      
      * lint x?
      
      * lint x??
      
      * fix
      
      * export symbols
      
      * skip gpu test
      
      * addresses comments
      
      * replaces another VecToIdArray
      
      * add randomwalks.h to include
      
      * replace void * with template
      a9dabcc7
  29. 19 Jan, 2020 1 commit
    • VoVAllen's avatar
      [Tensorflow] Several nn & example (#1191) · a00636a0
      VoVAllen authored
      * several nn example
      
      * appnp
      
      * fix lint
      
      * lint
      
      * add dgi
      
      * fix
      
      * fix
      
      * fix
      
      * fff
      
      * docs
      
      * 111
      
      * fix
      
      * change init
      
      * change result
      
      * tiaocan+1
      
      * fix
      
      * fix lint
      
      * fix
      
      * fix
      a00636a0
  30. 09 Jan, 2020 2 commits
  31. 06 Jan, 2020 1 commit
  32. 30 Dec, 2019 2 commits
    • Chao Ma's avatar
      [KVStore] Reduce memory cost of kvstore (#1156) · 6731ea3a
      Chao Ma authored
      * API change of kvstore
      
      * add demo for kvstore
      
      * update
      
      * remove duplicated log
      
      * change queue size
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * change name
      
      * update
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * change message queue size to a python argument
      
      * change default queue size to 2GB
      
      * OMP_NUM_THREADS=1
      
      * add multiple NICs support for kvstore
      
      * test
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      6731ea3a
    • Chao Ma's avatar
      [KVStore] Made kvstore can support multiple NICs (#1150) · 4f02bb75
      Chao Ma authored
      * API change of kvstore
      
      * add demo for kvstore
      
      * update
      
      * remove duplicated log
      
      * change queue size
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * change name
      
      * update
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * change message queue size to a python argument
      
      * change default queue size to 2GB
      
      * OMP_NUM_THREADS=1
      
      * add multiple NICs support for kvstore
      
      * test
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      4f02bb75
  33. 25 Dec, 2019 1 commit
    • Chao Ma's avatar
      [Small Fix] Change default message queue size of communicator to 2GB (#1140) · d57ff78d
      Chao Ma authored
      * API change of kvstore
      
      * add demo for kvstore
      
      * update
      
      * remove duplicated log
      
      * change queue size
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * change name
      
      * update
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * change message queue size to a python argument
      
      * change default queue size to 2GB
      
      * OMP_NUM_THREADS=1
      d57ff78d
  34. 24 Dec, 2019 1 commit
    • Chao Ma's avatar
      [KVStore] API change of kvstore (#1058) · e4ef8d1a
      Chao Ma authored
      * API change of kvstore
      
      * add demo for kvstore
      
      * update
      
      * remove duplicated log
      
      * change queue size
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix lint
      
      * change name
      
      * update
      
      * fix lint
      
      * update
      
      * update
      
      * update
      
      * update
      
      * change message queue size to a python argument
      e4ef8d1a
  35. 23 Dec, 2019 1 commit