XiaShan1227 / Graphormer

Do Transformers Really Perform Bad for Graph Representation? [NIPS-2021]
36 stars 8 forks source link
attention-mechanism deep-learning edge-embedding node-embedding self-attention transformer

【NIPS-2021 Graphormer】Do Transformers Really Perform Bad for Graph Representation? image

1.Environment

torch            2.0.1+cu118
torch_geometric  2.6.1

2.Run

python main.py --exp_name=ESOL_Graphormer

Code Framework Reference: Graphormer