GCS_aamas337
GCS_aamas337 copied to clipboard
Running issue
File "/home/zhangxi/code/GCS/onpolicy/algorithms/r_mappo/algorithm/graph_net_trans.py", line 22, in init self.GAT1 = GATConv(n_xdims, 8, heads=gat_nhead, concat=True, dropout=0.3) NameError: name 'GATConv' is not defined
Could you help me?
- Sorry for that, you can you the upper part
nn.TransformerEncoderLayer
. The GAT with the fully connected graph is just equal to the Transformer. - If you really want to use the
GATConv
, please import :from torch_geometric.nn import GCNConv
Sorry, I try again. But the function's arguments do not match.
- The GCNConv only has two arguments.
- The nn.TransformerEncoderLayer doesn't have the concatenate operation Can you show me the source code?