dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
Graph Attention Networks (GAT)
- Paper link: https://arxiv.org/abs/1710.10903
- Author's code repo: https://github.com/PetarV-/GAT.
Note that the original code is implemented with Tensorflow for the paper.
Dependencies
- MXNet nightly build
- requests
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