GAT implementation

#1

Hi there,

I have a concern about the implementation of GAT (https://github.com/dmlc/dgl/blob/master/examples/pytorch/gat/gat.py). It looks like when you create a GAT object, you must specify the graph in the constructor (https://github.com/dmlc/dgl/blob/master/examples/pytorch/gat/train.py), which means this object can only be used for this specific graph. But a lot of times we want this object to work on different graph inputs. How do you think?

Thanks

0 Likes

#2

Should that be the case, you can alternatively treat g as an argument to forward. This should work with the implementation we provide.

0 Likes

#3

Yes. I changed the code and it works.

0 Likes