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 |
|