Skip to content

Implementation of the graph attention network (GAT) paper in tinygrad

Notifications You must be signed in to change notification settings

ethanthoma/graph-attention-network

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Graph Attention Network

This is an implementation of the GAT paper in tinygrad.

The ./app/train.py script uses the CORA dataset.

The model gets 80.39% accuracy on the test set (2/3rds of the dataset) after 1000 epochs.

Full training time was 51:11 minutes on a surface pro 9, CPU only.

Checkpoints for every 100 epochs are found in ./models/ along with the final model.

About

Implementation of the graph attention network (GAT) paper in tinygrad

Topics

Resources

Stars

Watchers

Forks