mufeili/dgl_gat_nodeflow

This is a demo for training GATs on giant graphs with dgl nodeflow.

dgl_gat_nodeflow

This is a demo for training GATs on giant graphs with dgl nodeflow. We use GraphSAGE's reddit dataset for this example.

Dependencies

• dgl
  • You may need to install from source for the latest version.
• PyTorch
• scikit-learn
• tensorboardX
• TensorFlow (for the use of tensorboard)

Usage

python reddit.py

GPU will be used if available.

The training process can be monitored with tensorboard. To launch the tensorboard, do tensorboard --logdir=.