Graph Attention Networks (GAT)
Note that the original code is implemented with Tensorflow for the paper.
- MXNet nightly build
pip install mxnet --pre pip install requests
Usage (make sure that DGLBACKEND is changed into mxnet)
DGLBACKEND=mxnet python gat_batch.py --dataset cora --gpu 0 --num-heads 8