mirrors / dmlc--dgl

文件历史

610 提交 34 分支 42 Git 标签
0.3.x
提交图

11 次代码提交

作者 SHA1 备注 提交日期
VoVAllen 0f12763725 [Model] Early stop GAT (#750) 
* Add early stop

* add mxnet version

* Poke ci
 2019-08-21 14:58:56 +08:00
Lingfan Yu 74e13eea61 [Model] Update GAT model code (#622) 
* 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
 2019-06-08 21:41:38 -04:00
Chao Ma f99725adbc all demo use python-3 (#555) 2019-05-23 10:38:05 +08:00
Hao Zhang 039a711da1 [Model] fix self-edge bug in GCN and GAT. (#482) 
* Update gcn_mp.py

* Update train.py

* Update train.py

* Update train.py

* Update gat_batch.py

* Update gat_batch.py

* Update gcn_mp.py
 2019-04-10 09:04:19 -04:00
Giuseppe Futia 6c3dba867a [Tutorial] Update prerequisites of README (#380) 
* Update prerequisites of README

* dependencies for pytorch models

* dependencies for mxnet models

* minor
 2019-02-11 22:58:17 -05:00
Minjie Wang efae0f97d8 [Model] Improve GAT models (#348) 
* two better GAT implementations

* update numbers

* use version switch for spmm

* add missing dropout and output heads
 2019-01-11 02:02:06 -05:00
Hao Zhang 896dc50e70 [Model] fix bugs in pygat and mxgat. (#323) 
* mxgat

* pygat

* Update gat_batch.py

* add train

* Update gat_batch.py

* Update gat.py

* Update gat_batch.py
 2018-12-23 20:45:25 -05:00
Hao Zhang 632d598c77 [Doc] Add readme to mxnet gat (#300) 
* readme

* Update and rename README to README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md
 2018-12-17 04:36:26 -08:00
Da Zheng f8811c7d3c [BUGFIX] fix some minor problems in GAT (#308) 
* fix gat.

* fix context.
 2018-12-16 22:20:27 -08:00
Da Zheng bd0e4fa0b3 [MXNet][API] move to the new API (#123) 
* move gat to the new api.

* fix gcn.

* update sse.

* fix dgl core.

* update sse.

* fix small bugs in dgl core.

* fix mxnet tests.

* retrigger

* address comments and fix more bugs.

* fix

* fix tests.
 2018-11-08 10:00:47 -08:00
Da Zheng 5567f4a4d9 [BACKEND] Add MXNet backend. (#77) 
* support mxnet.

* add mxnet version of GCN.

* rename mxnet.nd as F.

* add mxnet GAT.

* enable GPU for GCN.

* fix MXNet GCN train.

* Use adam to optimize GAT

* support more operators.

* support sparse arrays.

* update mxnet backend.

* support index_copy.

* remove NN.

* update mxnet backend.

* temp check in.

* fix data conversion.

* add test.

* clean up mxnet backend.

* update mxnet examples.

* Revert "remove NN."

This reverts commit d815d9a0ec619f9ce9099c48cd35db9d8e947483.

* temp disable MXNet version of NN.
 2018-10-14 11:15:48 -04:00