1. 05 Mar, 2020 1 commit
    • 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
  2. 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
  3. 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
  4. 13 Feb, 2020 1 commit
  5. 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
  6. 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
  7. 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
  8. 30 Nov, 2019 1 commit
  9. 08 Nov, 2019 1 commit
  10. 05 Nov, 2019 1 commit
  11. 04 Nov, 2019 2 commits
  12. 03 Nov, 2019 1 commit
    • Zihao Ye's avatar
      [NN] nn modules & examples update (#890) · 9a0511c8
      Zihao Ye authored
      * upd
      
      * damn it
      
      * fuck
      
      * fuck pylint
      
      * fudge
      
      * remove some comments about MXNet
      
      * upd
      
      * upd
      
      * damn it
      
      * damn it
      
      * fuck
      
      * fuck
      
      * upd
      
      * upd
      
      * pylint bastard
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      9a0511c8
  13. 30 Oct, 2019 1 commit
    • xiang song(charlie.song)'s avatar
      [Bug Fix] Fix package reliability bug of networkx (#949) · 82499e60
      xiang song(charlie.song) authored
      * upd
      
      * fig edgebatch edges
      
      * add test
      
      * trigger
      
      * Update README.md for pytorch PinSage example.
      
      Add noting that the PinSage model example under
      example/pytorch/recommendation only work with Python 3.6+
      as its dataset loader depends on stanfordnlp package
      which work only with Python 3.6+.
      
      * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
      
      1. make dgl.nn.xxx frame agnostic
      2. make test.backend include dgl.nn modules
      3. modify test_edge_softmax of test/mxnet/test_nn.py and
          test/pytorch/test_nn.py work on both CPU and GPU
      
      * Fix style
      
      * Delete unused code
      
      * Make agnostic test only related to tests/backend
      
      1. clear all agnostic related code in dgl.nn
      2. make test_graph_conv agnostic to cpu/gpu
      
      * Fix code style
      
      * fix
      
      * doc
      
      * Make all test code under tests.mxnet/pytorch.test_nn.py
      work on both CPU and GPU.
      
      * Fix syntex
      
      * Remove rand
      
      * Add TAGCN nn.module and example
      
      * Now tagcn can run on CPU.
      
      * Add unitest for TGConv
      
      * Fix style
      
      * For pubmed dataset, using --lr=0.005 can achieve better acc
      
      * Fix style
      
      * Fix some descriptions
      
      * trigger
      
      * Fix doc
      
      * Add nn.TGConv and example
      
      * Fix bug
      
      * Update data in mxnet.tagcn test acc.
      
      * Fix some comments and code
      
      * delete useless code
      
      * Fix namming
      
      * Fix bug
      
      * Fix bug
      
      * Add test for mxnet TAGCov
      
      * Add test code for mxnet TAGCov
      
      * Update some docs
      
      * Fix some code
      
      * Update docs dgl.nn.mxnet
      
      * Update weight init
      
      * Fix
      
      * reproduce the bug
      
      * Fix concurrency bug reported at #755.
      Also make test_shared_mem_store.py more deterministic.
      
      * Update test_shared_mem_store.py
      
      * Update dmlc/core
      
      * networkx >= 2.4 will break our examples
      
      * Update tutorials/requirements
      
      * fix selfloop edges
      
      * upd version
      82499e60
  14. 21 Oct, 2019 1 commit
  15. 26 Sep, 2019 1 commit
    • Minjie Wang's avatar
      [Model][Hetero] GCMC using new hetero APIs (#860) · 9737b270
      Minjie Wang authored
      * init
      
      * rm data
      
      * README
      
      * fix rating values that are not decimal
      
      * rm stale codes
      
      * small fix
      
      * upd
      
      * rewrite decoder
      
      * fix many
      
      * many fix; performance matched
      
      * upd; handle sparse input
      
      * upd
      
      * address comments
      
      * more docstring; download data automatically
      
      * shared param mode
      9737b270
  16. 11 Sep, 2019 1 commit
  17. 09 Sep, 2019 1 commit
  18. 28 Aug, 2019 1 commit
    • xiang song(charlie.song)'s avatar
      [NN] Add MXNet impl for TAGCN module. (#799) · e17add56
      xiang song(charlie.song) authored
      * upd
      
      * fig edgebatch edges
      
      * add test
      
      * trigger
      
      * Update README.md for pytorch PinSage example.
      
      Add noting that the PinSage model example under
      example/pytorch/recommendation only work with Python 3.6+
      as its dataset loader depends on stanfordnlp package
      which work only with Python 3.6+.
      
      * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
      
      1. make dgl.nn.xxx frame agnostic
      2. make test.backend include dgl.nn modules
      3. modify test_edge_softmax of test/mxnet/test_nn.py and
          test/pytorch/test_nn.py work on both CPU and GPU
      
      * Fix style
      
      * Delete unused code
      
      * Make agnostic test only related to tests/backend
      
      1. clear all agnostic related code in dgl.nn
      2. make test_graph_conv agnostic to cpu/gpu
      
      * Fix code style
      
      * fix
      
      * doc
      
      * Make all test code under tests.mxnet/pytorch.test_nn.py
      work on both CPU and GPU.
      
      * Fix syntex
      
      * Remove rand
      
      * Add TAGCN nn.module and example
      
      * Now tagcn can run on CPU.
      
      * Add unitest for TGConv
      
      * Fix style
      
      * For pubmed dataset, using --lr=0.005 can achieve better acc
      
      * Fix style
      
      * Fix some descriptions
      
      * trigger
      
      * Fix doc
      
      * Add nn.TGConv and example
      
      * Fix bug
      
      * Update data in mxnet.tagcn test acc.
      
      * Fix some comments and code
      
      * delete useless code
      
      * Fix namming
      
      * Fix bug
      
      * Fix bug
      
      * Add test code for mxnet TAGCov
      
      * Update some docs
      
      * Fix some code
      
      * Update docs dgl.nn.mxnet
      
      * Update weight init
      
      * Fix
      e17add56
  19. 27 Aug, 2019 1 commit
  20. 23 Aug, 2019 1 commit
    • Minjie Wang's avatar
      [NN] RGCN modules (#744) · 708765f0
      Minjie Wang authored
      * rgcn module
      
      * support id input
      
      * WIP: model codes
      
      * use faster index select
      
      * dropout
      
      * self loop
      
      * WIP: link prediction
      
      * fix lint
      
      * WIP: docs
      
      * docstring
      
      * docstring
      
      * merge two child classes
      
      * mxnet rgcn module
      
      * fix lint
      
      * fix lint
      
      * fix rename bug
      
      * add uniform edge sampler
      
      * fix fn name
      
      * docstring
      
      * fix mxnet rgcn module
      
      * fix mx rgcn
      
      * enable test on cuda
      708765f0
  21. 21 Aug, 2019 1 commit
  22. 12 Jun, 2019 1 commit
  23. 10 Jun, 2019 1 commit
    • Da Zheng's avatar
      [BUGFIX] extand DGLGraph graph_data. (#632) · 411bcd7e
      Da Zheng authored
      * enable tutorial test in CI.
      
      * extand DGLGraph graph_data.
      
      * update doc.
      
      * Revert "enable tutorial test in CI."
      
      This reverts commit cd774067180922bb6ae979bde4aecbffc61c8147.
      
      * accept DGLGraph in graph store.
      411bcd7e
  24. 09 Jun, 2019 3 commits
    • Da Zheng's avatar
      Define node/edge Ids in NodeFlow more clearly (#628) · fc7775a2
      Da Zheng authored
      * add tests.
      
      * distinguish layer-local nid and nodeflow nid.
      
      * use numpy assert_array_equal and assert_allclose
      
      * fix map_from_parent_nid
      
      * fix test
      
      * fix test.
      
      * renmae remap.
      
      * update doc.
      
      * update doc.
      
      * update doc.
      
      * fix test.
      
      * fix test.
      fc7775a2
    • Minjie Wang's avatar
      [Examples] Run all the examples (#623) · dec8b49b
      Minjie Wang authored
      * all pytorch examples
      
      * scan through mxnet examples
      
      * change reddit data
      
      * tweak numerical range for unittest
      
      * fix ci
      
      * fix ci
      
      * fix
      
      * add seed to workaround
      dec8b49b
    • Lingfan Yu's avatar
      [Model] Update GAT model code (#622) · 74e13eea
      Lingfan Yu authored
      * fix gat code to use latest edge softmax module
      
      * avoid transpose
      
      * update README
      
      * use edge_softmax op
      
      * mxnet edge softmax op
      
      * mxnet gat
      
      * update README
      
      * fix unittest
      
      * fix ci
      
      * fix mxnet nn test; relax criteria for prod reducer
      74e13eea
  25. 02 Jun, 2019 1 commit
    • Minjie Wang's avatar
      [Graph] Add API to convert graph to simple graph (#587) · 01a4cc5b
      Minjie Wang authored
      * to simple
      
      * WIP: multigraph flag
      
      * graph index refactor; pass basic testing
      
      * graph index refactor; pass basic testing
      
      * fix bug in to_simple; pass torch test
      
      * fix mx utest
      
      * fix example
      
      * fix lint
      
      * fix ci
      
      * poke ci
      
      * poke ci
      
      * WIP
      
      * poke ci
      
      * poke ci
      
      * poke ci
      
      * change ci workspace
      
      * poke ci
      
      * poke ci
      
      * poke ci
      
      * poke ci
      
      * delete ci
      
      * use enum for multigraph flag
      01a4cc5b
  26. 30 May, 2019 2 commits
  27. 28 May, 2019 1 commit
  28. 26 May, 2019 1 commit
  29. 25 May, 2019 1 commit
  30. 23 May, 2019 4 commits
  31. 21 May, 2019 1 commit
    • Da Zheng's avatar
      [API] update graph store API. (#549) · b2b8be25
      Da Zheng authored
      * add init_ndata and init_edata in DGLGraph.
      
      * adjust SharedMemoryGraph API.
      
      * print warning.
      
      * fix comment.
      
      * update example
      
      * fix.
      
      * fix examples.
      
      * add unit tests.
      
      * add comments.
      b2b8be25
  32. 20 May, 2019 1 commit
    • Da Zheng's avatar
      [BUGFix] Improve multi-processing training (#526) · cdfca992
      Da Zheng authored
      * fix.
      
      * add comment.
      
      * remove.
      
      * temp fix.
      
      * initialize for shared memory.
      
      * fix graphsage.
      
      * fix gcn.
      
      * add more unit tests.
      
      * add more tests.
      
      * avoid creating shared-memory exclusively.
      
      * redefine remote initializer.
      
      * improve initializer.
      
      * fix unit test.
      
      * fix lint.
      
      * fix lint.
      
      * initialize data in the graph store server properly.
      
      * fix test.
      
      * fix test.
      
      * fix test.
      
      * small fix.
      
      * add comments.
      
      * cleanup server.
      
      * test graph store with a random port.
      
      * print.
      
      * print to stderr.
      
      * test1
      
      * test2
      
      * remove comment.
      
      * adjust the initializer signature.
      cdfca992