Zihao Ye
edf644633d
[performance] Optimize the association order of AXW in GraphSAGE. ( #2747 )
...
* upd
* lint
* upd
* upd
* compatibility
* upd
* upd
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-03-19 01:41:33 +08:00
Jinjing Zhou
366cc7ebf3
[Pickle] Fix HeteroGraphConv pickle problem ( #2761 )
...
* fix pickle problem
* lint
* add pickle tests
* fix
* fix
* fix
* fix
* fix for windows
2021-03-18 17:56:28 +08:00
Tianqi Zhang (张天启)
c88fca5055
[Feature] Add edge coarsening for homogeneous undirected graphs ( #2691 )
...
* finish graph matching gpu version
* use C++ shuffle
* finish graph matching
* fix bug
* fix bug
* change name and use swap
* upt
* fix format problem
* fix format problem
* stronger test
* upt
* upt
* change python api
* upt
* upt
* format check
* upt
* upt
* fix bug
Co-authored-by: Tong He <hetong007@gmail.com >
2021-03-09 16:35:27 +08:00
Mufei Li
7380d61eb9
[NN] Support Unidirectional Bipartite Graphs in CFConv ( #2674 )
...
* Update
* update
* Update
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-03-03 16:35:07 +08:00
Quan (Andy) Gan
d93a975997
revert ( #2672 )
2021-02-19 12:05:42 +08:00
Quan (Andy) Gan
9e630101a1
fix and lots of tests ( #2650 )
2021-02-12 11:40:05 +08:00
Quan (Andy) Gan
1f6eba9ed3
[Bug] Heterogeneous graph convolution bugfix ( #2578 )
...
* fix heterograph conv
* remove test cases
* fix test
* fix test
* fix test
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-01-29 09:30:19 +08:00
Chen Sirui
e4ddafe951
[NN] add multihead in DotGatConv ( #2549 )
...
* add multihead in DotGatConv
* Fix spacing issue
* Add Unit test for dotgat
* Modified Unit test for dotgat
* Add transformer like divisor
* Update dotgatconv.py
Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com >
Co-authored-by: Zihao Ye <expye@outlook.com >
2021-01-28 21:05:10 +08:00
xiang song(charlie.song)
a7e941c379
[Feature] Add support for sparse embedding ( #2451 )
...
* Add sparse embedding for dgl and update rgcn example
* upd
* Fix
* Revert "Fix"
This reverts commit 4da87cdfb8b8c3506b7fc7376cd2385ba8045c2a.
* Fix
* upd
* upd
* Fix
* Add unitest and update impl
* fix
* Clean up rgcn example code
* upd
* upd
* update
* Fix
* update score
* sparse for sage
* remove model sparse
* upd
* upd
* remove global norm
* revert delete model_sparse.py
* update according to comments
* Fix doc
* upd
* Fix test
* upd
* lint
* lint
* lint
* upd
* upd
* clean up
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal >
2021-01-28 00:26:49 +08:00
Tong He
0855d25501
[NN] Support scalar edge weight for GraphConv, SAGEConv and GINConv ( #2557 )
...
* add edge weight in forward
* fix lint
* fix
* fix
* address comments
* add utils
* add util to normalize in gcn way
* fix lint
* add unittest
* fix lint
* fix docstring
* fix docstring
* address comments
* improve notation consistence
* use preferred fn
2021-01-26 23:34:12 +08:00
Mufei Li
5da3439c16
[Feature] Add Device Flag in Data Loaders ( #2450 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-01-15 07:39:26 +08:00
xiang song(charlie.song)
54f186bba0
Fix ( #2524 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-59-204.ec2.internal >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-01-13 15:09:21 +08:00
Quan (Andy) Gan
bdb88df576
[Performance] Allow DataLoaders to return indices of the seed array ( #2495 )
...
* return indices from dataloader
* fixes
* fix
* fix distgraph and added some todos
* Update dataloader.py
* Update dataloader.py
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
2021-01-08 14:30:52 +08:00
Tong He
e2e524df41
[Feature] Add GraphDataLoader implementation ( #2496 )
...
* add graph dataloader
* add to doc
* fix
* fix
* fix docstring
* update according to torch default_collate
* add unittest
* fix
* fix lint
* fix
2021-01-07 13:38:21 +08:00
Minjie Wang
d3c1645550
[Performance][Test] New low memory implementation for RGCN and related regression tests ( #2468 )
...
* WIP
* finish lowmem impl; benchmarking
* wip
* wip
* fix benchmarks
* fix bug in searchsorted
* update readme
* update
* lint and ogb dependency
* add flags to to_homogeneous
* fix docstring
* address comments
2021-01-04 22:23:59 +08:00
Mufei Li
013d145663
[NN] Attention Retrieval for NN Modules ( #2397 )
...
* Update
* Update
2020-12-07 23:42:25 +08:00
Quan (Andy) Gan
ac5c79cdb8
[Bug] Fixes issues in feature pickling and transmission for subgraphs and blocks ( #2139 )
...
* bug fixes
* remove __deepcopy__; not sure what the behavior should be
* lint
* skip gpu test
* fix
* fix dist dataloader
* add comment
* remove assert
2020-09-03 12:48:59 +08:00
Mufei Li
9088c63592
[Feature] Change the Default Value of transpose in DGLGraph.adj ( #2046 )
...
* Update adj
* Fix
2020-08-18 22:45:46 +08:00
Mufei Li
be444e52d9
[Doc/Feature] Refactor, doc update and behavior fix for graphs ( #1983 )
...
* Update graph
* Fix for dgl.graph
* from_scipy
* Replace canonical_etypes with relations
* from_networkx
* Update for hetero_from_relations
* Roll back the change of canonical_etypes to relations
* heterograph
* bipartite
* Update doc
* Fix lint
* Fix lint
* Fix test cases
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* Update
* Fix test
* Fix
* Update
* Use DGLError
* Update
* Update
* Update
* Update
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* Update
* Fix
* Update
* Fix
* Fix
* Fix
* Update
* Fix
* Update
* Fix
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
* Fix
* Update
* Update
* Update
* Update
* Update
* Update
* rewrite sanity checks
* delete unnecessary checks
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Fix
* Fix
* Fix
* Update
* Fix
* Update
* Fix
* Fix
* Update
* Fix
* Update
* Fix
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
Co-authored-by: Quan Gan <coin2028@hotmail.com >
2020-08-18 04:26:29 +08:00
Tianjun Xiao
b8a4b446e5
[NN][Fix] fix some random test ( #2031 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-08-17 10:19:31 +08:00
Tianjun Xiao
4672c022a8
[hotfix] no in-degree check on sage ( #2014 )
...
* no degre check on sage
* fix pytest
* no in degree check for heteroconv
2020-08-13 13:11:55 +08:00
Tianjun Xiao
69f5869f3b
[Doc] NN doc refactor Conv Layers ( #1672 )
...
* go through gcn, relgcn
* fix tagconv formula
* fix doc in sageconv
* fix sgconv doc
* replace hat with tilde
* more comments on gmmconv
* fix agnnconv chebconv doc
* modify nnconv doc
* remove &
* add nn conv examples
* Rebase master
* More merge conflicts
* check homo
* add back self loop for some convs, check homo in tranform
* add example for denseconv
* add example and doc for dotgat and cfconv
* check in-degree for graphconv
* add language fix
* gconv address all comments
* another round of change based on api template
* change agnn
* go through agnn, appnp, atomic, cf, cheb, dense, gat, sage modules
* finish pytorch part of nn conv
* mxnet graphconv done
* tensorflow graphconv works
* add new modules into doc
* add comments to not split code
* refine doc
* resr
* more comments
* more fix
* finish conv and dense conv part api
* pylint fix
* fix pylink
* fix pylint
* more fix
* fix
* fix test fail because zere in degree
* fix test fail
* sage is not update for mxnet tf
Co-authored-by: Ubuntu <ubuntu@ip-172-31-0-81.us-east-2.compute.internal >
2020-08-12 18:47:19 +08:00
xiang song(charlie.song)
5e34ca8bb5
[Feature] Lazy copy ndata, edata to device ( #1986 )
...
* Lazy to device
* remove print
* lint
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* lint
* Fix
* Revert "Fix"
This reverts commit 615c9b8f80f5f6ee2ab43c849a22f0083deedf3b.
* Add test for frame lazy update
* disable tensorflow
* upd
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-08-12 13:04:47 +08:00
Quan (Andy) Gan
f5eb80d221
[Feature] Edge DataLoader for edge classification & link prediction ( #1828 )
...
* clean commit
* oops forgot the most important files
* use einsum
* copy feature from frontier to block
* Revert "copy feature from frontier to block"
This reverts commit 5224ec963eb6a3ef1b6ab74d8ecbd44e4e42f285.
* temp fix
* unit test
* fix
* revert jtnn
* lint
* fix win64
* docstring fixes and doc indexing
* revert einsum in sparse bidecoder
* fix some examples
* lint
* fix due to some tediousness in remove_edges
* addresses comments
* fix
* more jtnn fixes
* fix
2020-08-11 18:00:58 +08:00
Zihao Ye
18bfec24cf
[hotfix] Refactor edge softmax module ( #1967 )
...
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* upd
2020-08-07 23:50:12 +08:00
Minjie Wang
22167f7203
[Refactor] Enable new kernel in all message passing APIs ( #1953 )
...
* WIP: frame refactor
* new frame
* simple update_all builtin
* move all subgraph routines into the same file
* sddmm & spmm schedule; node & edge udf
* degree bucketing
* some tricky 0deg corner cases
* bug in frame append
* merge test_hetero_basics and test_basics
* some code rearange
* fix test_heterograph
* add mean spmm
* enable all builtin combinations
* pass gpu test
* pass pytorch tests
* wip
* fix some pt debugging codes
* fix bug in mxnet backward
* pass all mxnet utests
* passed tf tests
* docstring
* lint
* lint
* fix broadcasting bugs
* add warning and clamp for mean reducer
* add test for zero-degree mean
* address comments
* lint
* small fix
2020-08-07 15:40:25 +08:00
Zihao Ye
34a067eaa0
[bugfix] Fix the memory leak issue of Cluster GAT under 0.5 kernel and simplify the bipartite GAT. ( #1908 )
...
* uipd
* upd
* upd
* upd
* upd
2020-08-02 00:59:27 +08:00
Minjie Wang
44089c8b4d
[Refactor][Graph] Merge DGLGraph and DGLHeteroGraph ( #1862 )
...
* Merge
* [Graph][CUDA] Graph on GPU and many refactoring (#1791 )
* change edge_ids behavior and C++ impl
* fix unittests; remove utils.Index in edge_id
* pass mx and th tests
* pass tf test
* add aten::Scatter_
* Add nonzero; impl CSRGetDataAndIndices/CSRSliceMatrix
* CSRGetData and CSRGetDataAndIndices passed tests
* CSRSliceMatrix basic tests
* fix bug in empty slice
* CUDA CSRHasDuplicate
* has_node; has_edge_between
* predecessors, successors
* deprecate send/recv; fix send_and_recv
* deprecate send/recv; fix send_and_recv
* in_edges; out_edges; all_edges; apply_edges
* in deg/out deg
* subgraph/edge_subgraph
* adj
* in_subgraph/out_subgraph
* sample neighbors
* set/get_n/e_repr
* wip: working on refactoring all idtypes
* pass ndata/edata tests on gpu
* fix
* stash
* workaround nonzero issue
* stash
* nx conversion
* test_hetero_basics except update routines
* test_update_routines
* test_hetero_basics for pytorch
* more fixes
* WIP: flatten graph
* wip: flatten
* test_flatten
* test_to_device
* fix bug in to_homo
* fix bug in CSRSliceMatrix
* pass subgraph test
* fix send_and_recv
* fix filter
* test_heterograph
* passed all pytorch tests
* fix mx unittest
* fix pytorch test_nn
* fix all unittests for PyTorch
* passed all mxnet tests
* lint
* fix tf nn test
* pass all tf tests
* lint
* lint
* change deprecation
* try fix compile
* lint
* update METIDS
* fix utest
* fix
* fix utests
* try debug
* revert
* small fix
* fix utests
* upd
* upd
* upd
* fix
* upd
* upd
* upd
* upd
* upd
* trigger
* +1s
* [kernel] Use heterograph index instead of unitgraph index (#1813 )
* upd
* upd
* upd
* fix
* upd
* upd
* upd
* upd
* upd
* trigger
* +1s
* [Graph] Mutation for Heterograph (#1818 )
* mutation add_nodes and add_edges
* Add support for remove_edges, remove_nodes, add_selfloop, remove_selfloop
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
* upd
* upd
* upd
* fix
* [Transfom] Mutable transform (#1833 )
* add nodesy
* All three
* Fix
* lint
* Add some test case
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* fix
* triger
* Fix
* fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
* [Graph] Migrate Batch & Readout module to heterograph (#1836 )
* dgl.batch
* unbatch
* fix to device
* reduce readout; segment reduce
* change batch_num_nodes|edges to function
* reduce readout/ softmax
* broadcast
* topk
* fix
* fix tf and mx
* fix some ci
* fix batch but unbatch differently
* new checkk
* upd
* upd
* upd
* idtype behavior; code reorg
* idtype behavior; code reorg
* wip: test_basics
* pass test_basics
* WIP: from nx/ to nx
* missing files
* upd
* pass test_basics:test_nx_conversion
* Fix test
* Fix inplace update
* WIP: fixing tests
* upd
* pass test_transform cpu
* pass gpu test_transform
* pass test_batched_graph
* GPU graph auto cast to int32
* missing file
* stash
* WIP: rgcn-hetero
* Fix two datasety
* upd
* weird
* Fix capsuley
* fuck you
* fuck matthias
* Fix dgmg
* fix bug in block degrees; pass rgcn-hetero
* rgcn
* gat and diffpool fix
also fix ppi and tu dataset
* Tree LSTM
* pointcloud
* rrn; wip: sgc
* resolve conflicts
* upd
* sgc and reddit dataset
* upd
* Fix deepwalk, gindt and gcn
* fix datasets and sign
* optimization
* optimization
* upd
* upd
* Fix GIN
* fix bug in add_nodes add_edges; tagcn
* adaptive sampling and gcmc
* upd
* upd
* fix geometric
* fix
* metapath2vec
* fix agnn
* fix pickling problem of block
* fix utests
* miss file
* linegraph
* upd
* upd
* upd
* graphsage
* stgcn_wave
* fix hgt
* on unittests
* Fix transformer
* Fix HAN
* passed pytorch unittests
* lint
* fix
* Fix cluster gcn
* cluster-gcn is ready
* on fixing block related codes
* 2nd order derivative
* Revert "2nd order derivative"
This reverts commit 523bf6c249bee61b51b1ad1babf42aad4167f206.
* passed torch utests again
* fix all mxnet unittests
* delete some useless tests
* pass all tf cpu tests
* disable
* disable distributed unittest
* fix
* fix
* lint
* fix
* fix
* fix script
* fix tutorial
* fix apply edges bug
* fix 2 basics
* fix tutorial
Co-authored-by: yzh119 <expye@outlook.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-7-42.us-west-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal >
2020-07-28 14:30:41 +08:00
Quan (Andy) Gan
ec2e24be6d
[Feature] Slice dstnode features in NN modules ( #1838 )
...
* slice dstdata from srcdata within nn module
* a bunch of fixes
* add comment
* fix gcmc layer
* repr for blocks
* fix
* fix context
* fix
* do not copy internal columns
* docstring
2020-07-25 21:57:49 +08:00
Jinjing Zhou
53fc639201
[Torch] Fix inter-process communication for PyTorch ( #1858 )
...
* fix
* add comment
2020-07-24 18:15:24 +08:00
Zihao Ye
f25b1a064d
[Feature] Autograd of gspmm and gsddmm on PyTorch/MXNet/Tensorflow ( #1680 )
...
* init
* reverse(by minjie
* upd
* upd
* upd
* upd
* upd
* upd
* upd
* gpu
* upd
* upd
* upd
* upd
* udp
* upd
* upd
* imidiot
* fix
* upd
* upd
* upd
* udp
* upd
* upd
* fix
* udp
* upd
* upd
* upd
* upd
* upd
* fix
* remove redundency
* upd
* upd
* upd
* cache
* upd
* upd
* upd
* upd
* upd
* udp
* upd
* trigger
* upd
* fix
* upd
* unused import
* upd
* upd
2020-07-13 23:21:22 +08:00
Mufei Li
fb9d213848
Fix SAGEConv ( #1792 )
2020-07-12 13:32:14 +08:00
Quan (Andy) Gan
168a88e570
[Sampling] NodeDataLoader for node classification ( #1635 )
...
* neighbor sampler data loader first commit
* more commit
* nodedataloader
* fix
* update RGCN example
* update OGB
* fixes
* fix minibatch RGCN crashing with self loop
* reverting gatconv test code
* fix
* change to new solution that doesn't require tf dataloader
* fix
* lint
* fix
* fixes
* change doc
* fix docstring
* docstring fixes
* return seeds and input nodes from data loader
* fixes
* fix test
* fix windows build problem
* add pytorch wrapper
* fixes
* add pytorch wrapper
* add unit test
* add -1 support to sample_neighbors & fix docstrings
* docstring fix
* lint
* add minibatch rgcn evaluations
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Tong He <hetong007@gmail.com >
2020-07-02 16:31:05 +08:00
Quan (Andy) Gan
2f983abe64
fix knn graph bugs ( #1716 )
2020-06-28 17:31:38 +08:00
Axel Nilsson
b6c4deb4e9
[NN] ChebConv update ( #1460 )
...
* New version for the convolutional layer
* Minor changes
* Minor changes
* Update python/dgl/nn/pytorch/conv/chebconv.py
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com >
* Resolved variable miss-naming, import simplifying and raising warning
* Added dg_warnings instead of warnings.warn
* add doc
* upd
* upd
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com >
Co-authored-by: yzh119 <expye@outlook.com >
2020-06-24 15:59:44 +08:00
Tong He
3d47693b1f
[Op] Farthest Point Sampler in Cpp and CUDA ( #1630 )
...
* working framework without actual algorithm logic
* rename
* fix
* fps passes compilation
* correct algorithm
* add cuda implementation
* update random start
* before refactor
* pass compilation but cuda not working
* working
* code working, will add docstring
* add mxnet support
* update docstring
* update doc and test
* cpplint
* cpcplint
* pylint
* temporary fix
* fix for win64
* fix unitetest
* fix
* fix
* remove comment
* move to geometry package
* remove redundant include
* add docstrings and comments
* add proof
* add validity check
2020-06-22 00:52:20 +08:00
xiang song(charlie.song)
61f007c43b
[NN] Add low memory support for nn.RelGraphConv ( #1631 )
...
* Add low memory support for relgraphconv
y
* lint
* Add tf support
* Fix lint
* Fix mxnet
* Fix mxnet
* Fix
* minor fix
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-06-15 13:49:11 +08:00
Zihao Ye
af61e2fbb4
[Feature] Support nn modules for bipartite graphs. ( #1392 )
...
* init gat
* fix
* gin
* 7 nn modules
* rename & lint
* upd
* upd
* fix lint
* upd test
* upd
* lint
* shape check
* upd
* lint
* address comments
* update tensorflow
Co-authored-by: Quan Gan <coin2028@hotmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-03-29 16:48:38 +08:00
Minjie Wang
3efb5d8ecf
[NN] Add HeteroGraphConv module for cleaner module definition ( #1385 )
...
* 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
2020-03-27 18:47:06 +08:00
Quan (Andy) Gan
4af02022b2
[Bug] Multiple fixes ( #1374 )
...
* multiple fixes
* lint
* lint x2
2020-03-19 11:19:44 +08:00
Quan (Andy) Gan
a9520f71ce
[Model][Sampler] GraphSAGE model, bipartite graph conversion & remove edges API ( #1297 )
...
* 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
2020-03-08 02:38:29 +08:00
Zihao Ye
93ac29ce34
[Refactor] Unify DGLGraph, BatchedDGLGraph and DGLSubGraph ( #1216 )
...
* 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: Minjie Wang <minjie.wang@nyu.edu >
2020-03-05 18:03:57 +08:00
Mufei Li
828a5e5bc6
[DGL-LifeSci] Migration and Refactor ( #1226 )
...
* 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
2020-02-04 01:38:09 +08:00
Zihao Ye
ca302a13ac
[Feature] Add dgl.nn.*.Sequential for usability ( #1166 )
...
* upd
* upd
* upd
* upd
* lint
* upd
* upd
* upd
* upd
Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com >
2020-01-09 12:39:22 +08:00
Chao Ma
4f02bb751b
[KVStore] Made kvstore can support multiple NICs ( #1150 )
...
* 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
2019-12-30 15:17:31 +08:00
VoVAllen
dd65ee211e
[CI] Change tests for flexibility
...
* change ci image
* fix
* force bash
* fix
* fix python version
* fix
* fix
* fix
* update gpu
* cuda
* jenkins
* fix build sh
* fix
* Revert "fix"
This reverts commit 6b091914b3ef6b4300fa662ee375aa4b27944f5c.
* try fix
* fix
* Revert "fix"
This reverts commit e42c3035fa4974c6b197aaba0748f7b45113ddaa.
* try fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix tests
* try fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix ctx problem
* fix many tests
* fix typo
* add backend
* move to pytorch folder
* fix?
* fix ci
* try skip
* try false
* try?
* try?
* Revert to 7d9a52f
* fix various
* fix lint
* Revert Jenkinsfile
* revert env
* revert env
* address comment
* remove file
2019-12-16 14:54:37 +08:00
Zihao Ye
9a0511c8e9
[NN] nn modules & examples update ( #890 )
...
* 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
2019-11-03 17:29:42 -05:00
Zihao Ye
6a4b5ae98a
[Feature] Edge softmax on a subset of edges in the graph. ( #842 )
...
* upd
* add test
* fix
* upd
* merge
* hotfix
* upd
* fix
2019-09-09 13:44:56 +08:00
xiang song(charlie.song)
e17add5602
[NN] Add MXNet impl for TAGCN module. ( #799 )
...
* 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
2019-08-28 13:19:17 +08:00
Zihao Ye
9314aabd1f
[Refactor] Interface of nn modules ( #798 )
...
* refactor
* upd mpnn
2019-08-27 22:29:25 +08:00