README.md 547 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
Graph Attention Networks (GAT)
============

- Paper link: [https://arxiv.org/abs/1710.10903](https://arxiv.org/abs/1710.10903)
- Author's code repo:
  [https://github.com/PetarV-/GAT](https://github.com/PetarV-/GAT).

Note that the original code is implemented with Tensorflow for the paper.

10
11
12
### Dependencies
* MXNet nightly build
* requests
13

14
15
16
17
18
19
20
```bash
pip install mxnet --pre
pip install requests
```


### Usage (make sure that DGLBACKEND is changed into mxnet)
21
```bash
22
DGLBACKEND=mxnet python3 train.py --dataset cora --gpu 0 --num-heads 8
23
```