Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md
gat_batch.py

README.md

Graph Attention Networks (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