Looks like `PyG` has a [GAT model](https://pytorch-geometric.readthedocs.io/en/latest/generated/torch_geometric.nn.models.GAT.html#torch_geometric.nn.models.GAT) too, not sure why I didn't use this one from the beginning.
Looks like
PyGhas a GAT model too, not sure why I didn't use this one from the beginning.