mirrors / dmlc--dgl

文件历史

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

12 次代码提交

作者 SHA1 备注 提交日期
xiang song(charlie.song) 09ec6020ac [Example] rm gcn spmv (#2044) 
* rm gcn spmv

* remove useless test

* remove useless test

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
 2020-08-18 17:48:38 +08:00
Da Zheng 4ea42e3e08 fix gcn. (#469) 2019-04-05 10:56:09 -07:00
Minjie Wang 565f0c88fc [WIP] [NN] Refactor NN package (#406) 
* refactor graph conv

* docs & tests

* fix lint

* fix lint

* fix lint

* fix lint script

* fix lint

* Update

* Style fix

* Fix style

* Fix style

* Fix gpu case

* Fix for gpu case

* Hotfix edgesoftmax docs

* Handle repeated features

* Add docstring

* Set default arguments

* Remove dropout from nn.conv

* Fix

* add util fn for renaming

* revert gcn_spmv.py

* mx folder

* fix wierd bug

* fix mx

* fix lint
 2019-02-25 18:41:21 -05: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
Da Zheng 9cd0df7dde [Doc] Update README in MXNet GCN. (#373) 
* update readme.

* Revert "update readme."

This reverts commit cd5ea512059adadfcb22a615a2c42bba12f334a8.

* update.
 2019-02-02 21:59:07 -08:00
Minjie Wang ddc2faa547 [Bugfix] Fix gat residual bug (#355) 
* fix gat residual bug

* fix the residual addition; output heads; add some shape notations;

* minor

* fix the output head average

* add requests package in requirement
 2019-01-12 08:37:01 -05:00
Minjie Wang f35ac5443e [Graph][Model] Cache adj & inc; MX GCN (#307) 
* add cache to adj and incmat

* Fix bug in cached adj/inc

* mx gcn spmv runnable; acc debugging...

* fix bug in mx gcn that loss is not correctly calculated

* fix mx utest

* fix as requested

* use raw parameter tensors rather than dense layer

* fix dropout

* Add numbers in readme
 2018-12-16 22:51:01 -05:00
Ziyue Huang cffa4034f5 [Model] fix GCN (#305) 
* mxnet gcn spmv

* update readme

* fix gcn

* pytorch gcn

* update readme
 2018-12-16 12:44:41 -05:00
Da Zheng 78269cedfd [Doc] Update GCN README.md (#224) 
* update readme.

* update readme
 2018-12-04 12:07:05 -05:00
yifeim 6f4898a128 [Model][MXNet] gcn normalization and compare with mlp baselines (#196) 
* clean up pr-188 and resubmit

* address Da comments
 2018-12-02 19:17:37 -08:00
yifeim 02eb463ab1 report accuracy and fix training mask (#181) 2018-11-27 15:15:57 +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